Class ComputeVarsWithSharedModel
Inheritance
ComputeVarsWithSharedModel
Syntax
public class ComputeVarsWithSharedModel : DefaultVars, IVars, ITensorAllocator, IDisposable
Methods
Declaration
protected ComputeBuffer CreateComputeBufferForModelTensors(Layer layer, out long offsetIntoModelWeights)
Parameters
Type |
Name |
Description |
Layer |
layer |
|
Int64 |
offsetIntoModelWeights |
|
Returns
Type |
Description |
ComputeBuffer |
|
Dispose()
Declaration
public override void Dispose()
Overrides
Declaration
protected override Tensor[] PrepareLayerInputTensors(Model model, Layer layer, IOps ops)
Parameters
Returns
Overrides