Searched defs:BatchMatMul (Results 1 – 5 of 5) 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/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 820 TEST_F(OpLevelCostEstimatorTest, BatchMatMul) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_grad_test.cc | 252 Tensor BatchMatMul(const Tensor& x, bool ax, const Tensor& y, bool ay) { in BatchMatMul() function in tensorflow::__anon3cec93f10111::MathGradTest
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1199 TEST_F(OpTest, BatchMatMul) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 15681 func BatchMatMul(scope *Scope, x tf.Output, y tf.Output, optional ...BatchMatMulAttr) (output tf.Ou… func
|