| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_matmul.h | 28 namespace batch_matmul {
61 const int batch_dim0 = batch_matmul::broadcast_dim( in BatchMatMul()
63 const int batch_dim1 = batch_matmul::broadcast_dim( in BatchMatMul()
65 const int batch_dim2 = batch_matmul::broadcast_dim( in BatchMatMul()
68 const int lhs_ext0 = batch_matmul::extent(extended_lhs_shape, 0); in BatchMatMul()
69 const int lhs_ext1 = batch_matmul::extent(extended_lhs_shape, 1); in BatchMatMul()
70 const int lhs_ext2 = batch_matmul::extent(extended_lhs_shape, 2); in BatchMatMul()
71 const int rhs_ext0 = batch_matmul::extent(extended_rhs_shape, 0); in BatchMatMul()
72 const int rhs_ext1 = batch_matmul::extent(extended_rhs_shape, 1); in BatchMatMul()
73 const int rhs_ext2 = batch_matmul::extent(extended_rhs_shape, 2); in BatchMatMul()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 40 namespace batch_matmul { namespace
720 static TfLiteRegistration r = {batch_matmul::Init, batch_matmul::Free, in Register_BATCH_MATMUL_REF()
721 batch_matmul::Prepare, in Register_BATCH_MATMUL_REF()
722 batch_matmul::Eval<batch_matmul::kReference>}; in Register_BATCH_MATMUL_REF()
728 batch_matmul::Init, batch_matmul::Free, batch_matmul::Prepare, in Register_BATCH_MATMUL_GENERIC_OPTIMIZED()
729 batch_matmul::Eval<batch_matmul::kGenericOptimized>}; in Register_BATCH_MATMUL_GENERIC_OPTIMIZED()
|
| D | Android.bp | 66 "batch_matmul.cc",
|
| D | BUILD | 552 "batch_matmul.cc",
|
| /external/llvm-project/mlir/test/Dialect/Linalg/ |
| D | roundtrip.mlir | 660 linalg.batch_matmul ins(%a3, %b3: memref<?x?x?xf32>, memref<?x?x?xf32>)
662 linalg.batch_matmul ins(%ta3, %tb3: tensor<?x?x?xf32>, tensor<?x?x?xf32>)
664 %res1 = linalg.batch_matmul ins(%ta3, %tb3: tensor<?x?x?xf32>, tensor<?x?x?xf32>)
667 %res2 = linalg.batch_matmul ins(%ta3, %b3: tensor<?x?x?xf32>, memref<?x?x?xf32>)
673 // CHECK: linalg.batch_matmul
674 // CHECK: linalg.batch_matmul
675 // CHECK: linalg.batch_matmul
676 // CHECK: linalg.batch_matmul
|
| D | affine.mlir | 106 linalg.batch_matmul ins(%A, %B: memref<?x?x?xf32>, memref<?x?x?xf32>)
|
| D | vectorization.mlir | 38 linalg.batch_matmul
|
| D | invalid.mlir | 380 linalg.batch_matmul ins(%a3, %b3: memref<?x?x?xf32>, memref<?x?xf32>)
|
| D | loops.mlir | 1182 linalg.batch_matmul ins(%A, %B : memref<?x?x?xf32>, memref<?x?x?xf32>)
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgNamedStructuredOpsSpec.tc | 22 def batch_matmul(A: f32(Batch, M, K), B: f32(Batch, K, N)) -> (C: f32(Batch, M, N)) {
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | BUILD | 238 "optimized/batch_matmul.h",
453 "reference/batch_matmul.h",
573 "reference/batch_matmul.h",
|
| /external/tensorflow/tensorflow/lite/ |
| D | tflite_static.bp | 45 "kernels/batch_matmul.cc",
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 1350 auto* batch_matmul = new BatchMatMulOperator; in ConvertBatchMatMulOperator() local
1353 batch_matmul->adj_x = GetBoolAttr(node, "adj_x"); in ConvertBatchMatMulOperator()
1356 batch_matmul->adj_y = GetBoolAttr(node, "adj_y"); in ConvertBatchMatMulOperator()
1358 batch_matmul->inputs = {node.input(0), node.input(1)}; in ConvertBatchMatMulOperator()
1359 batch_matmul->outputs = {node.name()}; in ConvertBatchMatMulOperator()
1362 RetainTensorFlowNodeDef(node, batch_matmul); in ConvertBatchMatMulOperator()
1364 model->operators.emplace_back(batch_matmul); in ConvertBatchMatMulOperator()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 248 csinfo_.batch_matmul = "BatchMatMul"; in MklLayoutRewritePass()
400 rinfo_.push_back({csinfo_.batch_matmul, in MklLayoutRewritePass()
401 mkl_op_registry::GetMklOpName(csinfo_.batch_matmul), in MklLayoutRewritePass()
909 string batch_matmul; member
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | hlo-legalize-to-linalg.mlir | 1027 // CHECK: linalg.batch_matmul
1050 // CHECK: %[[DOT:.*]] = linalg.batch_matmul
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf.mlir | 1709 // CHECK: "tfl.batch_matmul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<10x15xf32>, ten…
1717 // CHECK: "tfl.batch_matmul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<2x10x15xf32>, t…
1725 // CHECK: "tfl.batch_matmul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<?x10x15xf32>, t…
|
| D | ops.mlir | 1386 // CHECK: "tfl.batch_matmul"(%arg0, %arg1)
1387 …%0 = "tfl.batch_matmul"(%arg0, %arg1) {adj_x = false, adj_y = true} : (tensor<1x4x384x32x!quant.un…
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 999 def TFL_BatchMatMulOp : TFL_Op<"batch_matmul", [
|
| /external/tensorflow/ |
| D | RELEASE.md | 4335 * Performance optimizations for `batch_matmul` on multi-core CPUs.
4407 * Improved multi-threaded performance of `batch_matmul`.
|