| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_custom_op.cpp | 61 struct BatchMatMul { struct
62 …s<DenseIndex, 3> dimensions(const Tensor<float, 3>& input1, const Tensor<float, 3>& input2) const { in dimensions()
91 Tensor<float, 3> result = tensor1.customOp(tensor2, BatchMatMul()); in test_custom_binary_op() argument
|
| D | cxx11_tensor_custom_op_sycl.cpp | 88 struct BatchMatMul { struct
89 DSizes<DenseIndex, 3> dimensions(const TensorType& input1, const TensorType& input2) const { in dimensions()
98 void eval(const TensorType& input1, const TensorType& input2, in eval()
|
| /external/armnn/src/backends/reference/workloads/ |
| D | BatchMatMulImpl.hpp | 16 class BatchMatMul { class
|
| D | BatchMatMulImpl.cpp | 15 BatchMatMul::BatchMatMul(const BatchMatMulDescriptor& params, in BatchMatMul() function in armnn::BatchMatMul
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | batch_matmul.h | 28 inline void BatchMatMul(const RuntimeShape& lhs_shape, const float* lhs_data, in BatchMatMul() function
115 inline void BatchMatMul(const RuntimeShape& lhs_shape, const int8_t* lhs_data, in BatchMatMul() function
271 inline void BatchMatMul(const FullyConnectedParams& params, in BatchMatMul() function
377 inline void BatchMatMul(const FullyConnectedParams& params, in BatchMatMul() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_matmul.h | 54 inline void BatchMatMul(const RuntimeShape& lhs_shape, const Ta* lhs_data, in BatchMatMul() function
109 inline void BatchMatMul(const RuntimeShape& lhs_shape, const int8_t* lhs_data, in BatchMatMul() function
200 inline void BatchMatMul(const FullyConnectedParams& params, in BatchMatMul() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision_test.cc | 1104 TEST_F(AutoMixedPrecisionTest, BatchMatMul) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_grad_test.cc | 276 Tensor BatchMatMul(const Tensor& x, bool ax, const Tensor& y, bool ay) { in BatchMatMul() function in tensorflow::__anon5d8d1aad0111::MathGradTest
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1852 TEST_F(OpTest, BatchMatMul) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 2545 func BatchMatMul(scope *Scope, x tf.Output, y tf.Output, optional ...BatchMatMulAttr) (output tf.Ou… func
|