Searched defs:BatchMatMul (Results 1 – 7 of 7) sorted by relevance
| /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
|
| /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
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_matmul.h | 53 inline void BatchMatMul(const RuntimeShape& lhs_shape, const float* lhs_data, in BatchMatMul() function
108 inline void BatchMatMul(const RuntimeShape& lhs_shape, const int8_t* lhs_data, in BatchMatMul() function
199 inline void BatchMatMul(const FullyConnectedParams& params, in BatchMatMul() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision_test.cc | 1039 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::__anoncd3cd74f0111::MathGradTest
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1200 TEST_F(OpTest, BatchMatMul) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 27824 func BatchMatMul(scope *Scope, x tf.Output, y tf.Output, optional ...BatchMatMulAttr) (output tf.Ou… func
|