Searched refs:tensor_shape0 (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 101 TensorShape tensor_shape0 = input_tensor_0.shape(); in Compute() local
103 if (tf_shape0 == tf_shape1 && tensor_shape0 == tensor_shape1) { in Compute()
162 (tensor_shape0 == tensor_shape1)); in Compute()
|
| /external/ComputeLibrary/src/core/GLES_COMPUTE/kernels/ |
| D | GCGEMMMatrixMultiplyKernel.cpp | 81 TensorShape tensor_shape0{ input0->tensor_shape() }; in validate_arguments() local
82 tensor_shape0.set(0, k); in validate_arguments()
83 tensor_shape0.set(1, m); in validate_arguments()
89 const TensorInfo tensor_info0 = input0->clone()->set_tensor_shape(tensor_shape0); in validate_arguments()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLGEMMLowpMatrixMultiplyReshapedKernel.cpp | 68 TensorShape tensor_shape0{ input0->tensor_shape() }; in validate_arguments() local
69 tensor_shape0.set(0, k); in validate_arguments()
70 tensor_shape0.set(1, m); in validate_arguments()
76 const TensorInfo tensor_info0 = input0->clone()->set_tensor_shape(tensor_shape0); in validate_arguments()
|
| D | CLGEMMMatrixMultiplyReshapedKernel.cpp | 88 TensorShape tensor_shape0{ input0->tensor_shape() }; in validate_arguments() local
89 tensor_shape0.set(0, k); in validate_arguments()
90 tensor_shape0.set(1, m); in validate_arguments()
112 const TensorInfo tensor_info0 = input0->clone()->set_tensor_shape(tensor_shape0); in validate_arguments()
|
| D | CLGEMMMatrixMultiplyKernel.cpp | 110 TensorShape tensor_shape0{ input0->tensor_shape() }; in validate_arguments() local
111 tensor_shape0.set(0, k); in validate_arguments()
112 tensor_shape0.set(1, m); in validate_arguments()
118 const TensorInfo tensor_info0 = input0->clone()->set_tensor_shape(tensor_shape0); in validate_arguments()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEGEMMMatrixMultiplyKernel.cpp | 1054 TensorShape tensor_shape0{ input0->tensor_shape() }; in validate_arguments() local
1055 tensor_shape0.set(0, k); in validate_arguments()
1056 tensor_shape0.set(1, m); in validate_arguments()
1058 const TensorInfo tensor_info0 = input0->clone()->set_tensor_shape(tensor_shape0); in validate_arguments()
|