| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | batchmatmul_to_einsum.mlir | 6 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<1x2x3xf32>, tensor<3x…
13 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<2x2x4xf32>, tensor<2x…
20 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<1x2x?xf32>, tensor<?x…
27 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<1x2x?xf32>, tensor<3x…
35 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = true} : (tensor<1x2x3xf32>, tensor<4x3…
|
| D | unroll-batch-matmul.mlir | 115 …%0 = "tf.BatchMatMul"(%arg0, %arg1) : (tensor<2x3x4x5xf32>, tensor<2x3x5x6xf32>) -> tensor<2x3x4x6…
177 %0 = "tf.BatchMatMul"(%arg0, %arg1) : (tensor<3x4x5xf32>, tensor<3x5x6xf32>) -> tensor<3x4x6xf32>
215 %0 = "tf.BatchMatMul"(%arg0, %arg1) : (tensor<4x5xf32>, tensor<5x6xf32>) -> tensor<4x6xf32>
|
| D | canonicalize.mlir | 22 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<2x3x5xf32>, tensor<2x…
28 // CHECK: tf.BatchMatMul
29 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<2x3x5xf32>, tensor<?x…
35 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<2x3xf32>, tensor<3x2x…
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BatchMatMul.pbtxt | 2 name: "BatchMatMul"
45 name: "BatchMatMul"
89 name: "BatchMatMul"
133 name: "BatchMatMul"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BatchMatMul.pbtxt | 2 name: "BatchMatMul"
45 name: "BatchMatMul"
89 name: "BatchMatMul"
133 name: "BatchMatMul"
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_BatchMatMul.pbtxt | 2 graph_op_name: "BatchMatMul"
4 name: "linalg.BatchMatMul"
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_custom_op.cpp | 61 struct BatchMatMul { struct
91 Tensor<float, 3> result = tensor1.customOp(tensor2, BatchMatMul()); in test_custom_binary_op() argument
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_BatchMatMul.pbtxt | 2 graph_op_name: "BatchMatMul"
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 492 optimized_ops::BatchMatMul( in EvalHybrid()
498 reference_ops::BatchMatMul( in EvalHybrid()
528 reference_ops::BatchMatMul<int8_t, int32_t>( in EvalInt8()
533 optimized_ops::BatchMatMul(op_params, rhs_shape, GetTensorData<int8_t>(rhs), in EvalInt8()
562 reference_ops::BatchMatMul<int16_t, int64_t>( in EvalInt16()
691 optimized_ops::BatchMatMul(rhs_shape, GetTensorData<float>(rhs_tensor), in Eval()
697 reference_ops::BatchMatMul(rhs_shape, GetTensorData<float>(rhs_tensor), in Eval()
|
| /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/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/cc/client/ |
| D | client_session_test.cc | 36 using ops::BatchMatMul;
212 auto c = BatchMatMul(root, a, b); in TEST()
|
| /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
1190 test::ExpectClose(dx, BatchMatMul(dz, false, y, true)); in TEST_F()
1191 test::ExpectClose(dy, BatchMatMul(x, true, dz, false)); in TEST_F()
1202 test::ExpectClose(dx, BatchMatMul(dz, false, y, false)); in TEST_F()
1203 test::ExpectClose(dy, BatchMatMul(dz, true, x, false)); in TEST_F()
1214 test::ExpectClose(dx, BatchMatMul(y, false, dz, true)); in TEST_F()
1215 test::ExpectClose(dy, BatchMatMul(x, false, dz, false)); in TEST_F()
1226 test::ExpectClose(dx, BatchMatMul(y, true, dz, true)); in TEST_F()
1227 test::ExpectClose(dy, BatchMatMul(dz, true, x, true)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchMatMul.pbtxt | 2 graph_op_name: "BatchMatMul"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | canonicalize.td | 55 // BatchMatMul op patterns.
58 // Static shaped operands in a legal BatchMatMul op will have matching batch
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad_test.cc | 33 using ops::BatchMatMul;
611 z = BatchMatMul(root_, x, y, BatchMatMul::AdjX(t_x).AdjY(t_y)); in TestMatMulGrad()
|
| D | math_grad.cc | 1094 BatchMatMul(scope, x0, x1, BatchMatMul::AdjX(adj_x0).AdjY(adj_x1)); in MatMulGradHelper()
1097 BatchMatMul(scope, y0, y1, BatchMatMul::AdjX(adj_y0).AdjY(adj_y1)); in MatMulGradHelper()
|
| D | nn_grad.cc | 101 BatchMatMul(scope, grad_grad_expand, logits_softmax_expand); in SoftmaxCrossEntropyWithLogitsGrad()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_patterns.td | 280 // When batch size is known, TF BatchMatMul gets unfolded to TFL FullyConnected
282 // fall through to here and convert to TF Lite BatchMatMul.
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 56 * `raw_ops.BatchMatMul`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 31 `BatchMatMul` | `T={complex64,double,float,int32}`
|
| D | cpu_supported_ops.md | 31 `BatchMatMul` | `T={complex64,double,float,int32}`
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision_test.cc | 1039 TEST_F(AutoMixedPrecisionTest, BatchMatMul) { in TEST_F() argument
1042 Output allow1 = ops::BatchMatMul(s.WithOpName("allow1"), input, input); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema.fbs | 1014 // Parameters for BatchMatMul version 4 or above.
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1326 ### tf.BatchMatMul
1333 %output = tf.BatchMatMul(%x, %y, %adj_x, %adj_y)
|