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