Home
last modified time | relevance | path

Searched refs:BatchMatMul (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_BatchMatMul.pbtxt2 graph_op_name: "BatchMatMul"
4 name: "linalg.BatchMatMul"
/external/eigen/unsupported/test/
Dcxx11_tensor_custom_op.cpp61 struct BatchMatMul { struct
91 Tensor<float, 3> result = tensor1.customOp(tensor2, BatchMatMul()); in test_custom_binary_op() argument
/external/tensorflow/tensorflow/core/kernels/
Dbatch_matmul_op_impl.h533 class BatchMatMul : public OpKernel {
535 explicit BatchMatMul(OpKernelConstruction* context) : OpKernel(context) {
540 virtual ~BatchMatMul() {}
604 BatchMatMul<CPUDevice, TYPE>)
609 BatchMatMul<GPUDevice, TYPE>)
615 BatchMatMul<SYCLDevice, TYPE>)
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_BatchMatMul.pbtxt2 graph_op_name: "BatchMatMul"
/external/tensorflow/tensorflow/core/ops/
Dmath_grad_test.cc252 Tensor BatchMatMul(const Tensor& x, bool ax, const Tensor& y, bool ay) { in BatchMatMul() function in tensorflow::__anona51524880111::MathGradTest
1141 test::ExpectClose(dx, BatchMatMul(dz, false, y, true)); in TEST_F()
1142 test::ExpectClose(dy, BatchMatMul(x, true, dz, false)); in TEST_F()
1153 test::ExpectClose(dx, BatchMatMul(dz, false, y, false)); in TEST_F()
1154 test::ExpectClose(dy, BatchMatMul(dz, true, x, false)); in TEST_F()
1165 test::ExpectClose(dx, BatchMatMul(y, false, dz, true)); in TEST_F()
1166 test::ExpectClose(dy, BatchMatMul(x, false, dz, false)); in TEST_F()
1177 test::ExpectClose(dx, BatchMatMul(y, true, dz, true)); in TEST_F()
1178 test::ExpectClose(dy, BatchMatMul(dz, true, x, true)); in TEST_F()
Dops.pbtxt3307 name: "BatchMatMul"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BatchMatMul.pbtxt2 graph_op_name: "BatchMatMul"
/external/tensorflow/tensorflow/cc/gradients/
Dmath_grad.cc1061 BatchMatMul(scope, x0, x1, BatchMatMul::AdjX(adj_x0).AdjY(adj_x1)); in MatMulGradHelper()
1064 BatchMatMul(scope, y0, y1, BatchMatMul::AdjX(adj_y0).AdjY(adj_y1)); in MatMulGradHelper()
Dmath_grad_test.cc33 using ops::BatchMatMul;
589 z = BatchMatMul(root_, x, y, BatchMatMul::AdjX(t_x).AdjY(t_y)); in TestMatMulGrad()
Dnn_grad.cc101 BatchMatMul(scope, grad_grad_expand, logits_softmax_expand); in SoftmaxCrossEntropyWithLogitsGrad()
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md31 `BatchMatMul` | `T={complex64,double,float,int32}`
Dcpu_supported_ops.md31 `BatchMatMul` | `T={complex64,double,float,int32}`
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc820 TEST_F(OpLevelCostEstimatorTest, BatchMatMul) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc315 HANDLE_OPERATORTYPENAME_CASE(BatchMatMul) in OperatorTypeName()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc806 Output matmul = ops::BatchMatMul(matmul_op, trans_a, trans_b); in TEST_F()
862 Output matmul = ops::BatchMatMul(s.WithOpName("matmul"), trans_a, trans_b); in TEST_F()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1199 TEST_F(OpTest, BatchMatMul) { in TEST_F() argument
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt4610 name: "BatchMatMul"
4651 name: "BatchMatMul"
4692 name: "BatchMatMul"
4734 name: "BatchMatMul"
Dops_history.v1.pbtxt9222 name: "BatchMatMul"
9265 name: "BatchMatMul"
9309 name: "BatchMatMul"
9353 name: "BatchMatMul"
Dops_history.v2.pbtxt9222 name: "BatchMatMul"
9265 name: "BatchMatMul"
9309 name: "BatchMatMul"
9353 name: "BatchMatMul"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt316 name: "BatchMatMul"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt316 name: "BatchMatMul"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go15681 func BatchMatMul(scope *Scope, x tf.Output, y tf.Output, optional ...BatchMatMulAttr) (output tf.Ou… func