Documentation
¶
Index ¶
- func GetLayerTestResult(inPyScript string, layer torch.LayerForTesting, inTensor *tensor.Tensor) *tensor.Tensor
- func GetLayerTestResult32(inPyScript string, layer torch.LayerForTesting, inTensor *tensor.Tensor) *tensor.Tensor
- func GetLayerTestResult64(inPyScript string, layer torch.LayerForTesting, inTensor *tensor.Tensor, ...) *tensor.Tensor
- func GetPytorchInitData(pyScript string) *tensor.Tensor
- func GetTensorTestResult(inPyScript string, inTensor1 *tensor.Tensor, inTensor2 *tensor.Tensor) *tensor.Tensor
- func IsTesting() bool
- func LayerTest(inPyScript string, inTensor *tensor.Tensor, dataType string) *tensor.Tensor
- func LayerTest1(inPyScript string, inTensors []*tensor.Tensor, dataType string) *tensor.Tensor
- func RunPyScript(pythonScript string)
- func TimeMeasure(start time.Time) string
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func GetLayerTestResult ¶
func GetLayerTestResult32 ¶
func GetLayerTestResult64 ¶
func GetPytorchInitData ¶
func GetTensorTestResult ¶
func LayerTest1 ¶
func RunPyScript ¶
func RunPyScript(pythonScript string)
func TimeMeasure ¶
Types ¶
This section is empty.
Source Files
Click to show internal directories.
Click to hide internal directories.