Searched refs:tensor_shape1 (Results 1 – 8 of 8) sorted by relevance
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuGemmMatrixMultiplyKernel.cpp | 103 TensorShape tensor_shape1{ rhs->tensor_shape() }; in validate_arguments() local
104 tensor_shape1.set(0, n); in validate_arguments()
105 tensor_shape1.set(1, k); in validate_arguments()
107 const TensorInfo tensor_info1 = rhs->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 89 TensorShape tensor_shape1 = input_tensor_1.shape(); in Compute() local
90 if (tf_shape0 == tf_shape1 && tensor_shape0 == tensor_shape1) { in Compute()
150 (tensor_shape0 == tensor_shape1)); in Compute()
|
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClGemmLowpMatrixMultiplyReshapedKernel.cpp | 79 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
80 tensor_shape1.set(0, n); in validate_arguments()
81 tensor_shape1.set(1, k); in validate_arguments()
84 const TensorInfo tensor_info1 = src1->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
|
| D | ClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp | 127 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
128 tensor_shape1.set(0, n); in validate_arguments()
129 tensor_shape1.set(1, k); in validate_arguments()
131 const TensorInfo tensor_info1 = src1->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
|
| D | ClGemmMatrixMultiplyReshapedKernel.cpp | 107 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
108 tensor_shape1.set(0, n); in validate_arguments()
109 tensor_shape1.set(1, k); in validate_arguments()
128 const TensorInfo tensor_info1 = src1->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
|
| D | ClGemmLowpMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp | 72 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
73 tensor_shape1.set(0, n); in validate_arguments()
74 tensor_shape1.set(1, k); in validate_arguments()
76 const TensorInfo tensor_info1 = src1->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
|
| D | ClGemmMatrixMultiplyReshapedOnlyRhsKernel.cpp | 93 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
94 tensor_shape1.set(0, n); in validate_arguments()
95 tensor_shape1.set(1, k); in validate_arguments()
113 const TensorInfo tensor_info1 = src1->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
|
| D | ClGemmLowpMatrixMultiplyReshapedOnlyRhsKernel.cpp | 87 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
88 tensor_shape1.set(0, n); in validate_arguments()
89 tensor_shape1.set(1, k); in validate_arguments()
91 const TensorInfo tensor_info1 = src1->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
|