Home
last modified time | relevance | path

Searched refs:inputYTensorInfo (Results 1 – 6 of 6) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/
DBatchMatMulEndToEndTestImpl.hpp29 TensorInfo inputYTensorInfo(inputYShape, DataType, qScale, qOffset, true); in CreateBatchMatMulNetwork() local
43 Connect(inputY, batchMatMul, inputYTensorInfo, 0, 1); in CreateBatchMatMulNetwork()
DAdditionEndToEndTestImpl.hpp29 TensorInfo inputYTensorInfo(inputYShape, DataType, qScale, qOffset, true); in CreateAdditionNetwork() local
41 Connect(inputY, addition, inputYTensorInfo, 0, 1); in CreateAdditionNetwork()
DMultiplicationEndToEndTestImpl.hpp29 TensorInfo inputYTensorInfo(inputYShape, DataType, qScale, qOffset, true); in CreateMultiplicationNetwork() local
41 Connect(inputY, multiplication, inputYTensorInfo, 0, 1); in CreateMultiplicationNetwork()
DSubtractionEndToEndTestImpl.hpp29 TensorInfo inputYTensorInfo(inputYShape, DataType, qScale, qOffset, true); in CreateSubtractionNetwork() local
41 Connect(inputY, subtraction, inputYTensorInfo, 0, 1); in CreateSubtractionNetwork()
/external/armnn/src/backends/cl/workloads/
DClBatchMatMulWorkload.cpp147 arm_compute::TensorShape inputYTensorInfo = armcomputetensorutils::BuildArmComputeTensorShape( in ClBatchMatMulWorkload() local
149 inputY.info()->set_tensor_shape(inputYTensorInfo); in ClBatchMatMulWorkload()
/external/armnn/src/armnnTfLiteParser/
DTfLiteParser.cpp1717 TensorInfo inputYTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 1); in ParseBatchMatMul() local