/external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/ |
D | back2back_fake_quant.pbtxt | 31 name: "sequential/quant_dense/MatMul/ReadVariableOp/resource" 58 name: "sequential/quant_dense/MatMul/ReadVariableOp" 60 input: "sequential/quant_dense/MatMul/ReadVariableOp/resource" 69 name: "sequential/quant_dense/MatMul/kquant/FakeQuantWithMinMaxVars/ReadVariableOp/resource" 90 name: "sequential/quant_dense/MatMul/kquant/FakeQuantWithMinMaxVars/ReadVariableOp" 92 input: "sequential/quant_dense/MatMul/kquant/FakeQuantWithMinMaxVars/ReadVariableOp/resource" 101 name: "sequential/quant_dense/MatMul/kquant/FakeQuantWithMinMaxVars/ReadVariableOp_1/resource" 122 name: "sequential/quant_dense/MatMul/kquant/FakeQuantWithMinMaxVars/ReadVariableOp_1" 124 input: "sequential/quant_dense/MatMul/kquant/FakeQuantWithMinMaxVars/ReadVariableOp_1/resource" 133 name: "sequential/quant_dense/MatMul/kquant/FakeQuantWithMinMaxVars" [all …]
|
D | control_flow_v1.pbtxt | 118 name: "cond/MatMul" 119 op: "MatMul" 120 input: "cond/MatMul/Switch:1" 121 input: "cond/MatMul/Switch_1:1" 142 name: "cond/MatMul/Switch" 162 name: "cond/MatMul/Switch_1" 237 input: "cond/MatMul"
|
D | unroll_batch_matmul.pbtxt | 1 …hapes=2,5,3:3,7 -tf-input-data-types=DT_FLOAT,DT_FLOAT -tf-output-arrays=MatMul -output-mlir %s -o… 53 name: "MatMul" 80 …ntry_function = {control_outputs = "", inputs = "Placeholder,Placeholder_1", outputs = "MatMul"}} {
|
/external/tensorflow/tensorflow/cc/framework/ |
D | gradients_test.cc | 34 using ops::MatMul; 93 auto z = MatMul(scope, x, y); in TEST_F() 100 auto dx = MatMul(scope, dz, y, MatMul::TransposeB(true)); in TEST_F() 101 auto dy = MatMul(scope, x, dz, MatMul::TransposeA(true)); in TEST_F() 119 auto z = MatMul(scope, x, y); in TEST_F() 128 auto dx = MatMul(scope, dz, y, MatMul::TransposeB(true)); in TEST_F() 129 auto dy = MatMul(scope, x, dz, MatMul::TransposeA(true)); in TEST_F() 145 auto x = MatMul(scope, u, v); in TEST_F() 148 auto z = MatMul(scope, x, y); in TEST_F() 156 auto dx = MatMul(scope, dz, y, MatMul::TransposeB(true)); in TEST_F() [all …]
|
D | cc_ops_test.cc | 30 auto m = MatMul(cop_scopes.child, x, w); in Linear() 50 MatMul m(root, c, {{41}, {1}}); in TEST() 59 auto m = MatMul(root, {{1}, {1}}, {{41}, {1}}, MatMul::TransposeA(true)); in TEST() 104 auto y = MatMul(root.WithOpName("y"), a, x); in TEST()
|
/external/tensorflow/tensorflow/compiler/jit/tests/ |
D | opens2s_gnmt_mixed_precision.golden_summary | 119 MatMul 1 128 MatMul 2 190 MatMul 10 221 MatMul 20 241 MatMul 1 255 MatMul 2 281 MatMul 1 295 MatMul 2 311 MatMul 1 324 MatMul 1 [all …]
|
/external/llvm-project/llvm/lib/Transforms/Scalar/ |
D | LowerMatrixIntrinsics.cpp | 1091 CallInst *MatMul) { in getNonAliasingPointer() argument 1105 BasicBlock *Check0 = MatMul->getParent(); in getNonAliasingPointer() 1113 BasicBlock *Check1 = SplitBlock(MatMul->getParent(), MatMul, nullptr, LI, in getNonAliasingPointer() 1116 SplitBlock(MatMul->getParent(), MatMul, nullptr, LI, nullptr, "copy"); in getNonAliasingPointer() 1117 BasicBlock *Fusion = SplitBlock(MatMul->getParent(), MatMul, nullptr, LI, in getNonAliasingPointer() 1123 IRBuilder<> Builder(MatMul); in getNonAliasingPointer() 1170 bool isFusionProfitable(CallInst *MatMul) { in isFusionProfitable() argument 1174 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in isFusionProfitable() 1175 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in isFusionProfitable() 1180 auto *EltType = cast<VectorType>(MatMul->getType())->getElementType(); in isFusionProfitable() [all …]
|
/external/tensorflow/tensorflow/c/eager/ |
D | mnist_gradients_testutil.cc | 78 TF_RETURN_IF_ERROR(ops::MatMul(tape_ctx.get(), inputs, in MatMulGradModel() 122 TF_RETURN_IF_ERROR(ops::MatMul(ctx, {X, W1}, absl::MakeSpan(temp_outputs), in MNISTForwardModel() 131 TF_RETURN_IF_ERROR(ops::MatMul( in MNISTForwardModel() 156 TF_RETURN_IF_ERROR(ops::MatMul(ctx, {X, W1}, outputs, "matmul0", in MatMulTransposeModel() 177 TF_RETURN_IF_ERROR(ops::MatMul(tape_ctx.get(), {X, W1}, in MNISTGradModel() 190 TF_RETURN_IF_ERROR(ops::MatMul( in MNISTGradModel() 227 return ops::MatMul(ctx, inputs, outputs, "matmul0", in MatMulModel()
|
/external/tensorflow/tensorflow/c/experimental/gradients/ |
D | math_grad.cc | 28 using tensorflow::ops::MatMul; 151 TF_RETURN_IF_ERROR(MatMul(ctx, {upstream_grad, B.get()}, in Compute() 157 TF_RETURN_IF_ERROR(MatMul(ctx, {A.get(), upstream_grad}, in Compute() 163 TF_RETURN_IF_ERROR(MatMul(ctx, {upstream_grad, B.get()}, in Compute() 169 TF_RETURN_IF_ERROR(MatMul(ctx, {upstream_grad, A.get()}, in Compute() 176 TF_RETURN_IF_ERROR(MatMul(ctx, {B.get(), upstream_grad}, in Compute() 182 TF_RETURN_IF_ERROR(MatMul(ctx, {A.get(), upstream_grad}, in Compute() 188 TF_RETURN_IF_ERROR(MatMul(ctx, {B.get(), upstream_grad}, in Compute() 194 TF_RETURN_IF_ERROR(MatMul(ctx, {upstream_grad, A.get()}, in Compute()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
D | unroll-batch-matmul.mlir | 50 …// CHECK: %[[v26:.*]] = "tf.MatMul"(%[[v2]], %[[v15]]) {transpose_a = false, transpose_b = false} … 51 …// CHECK: %[[v27:.*]] = "tf.MatMul"(%[[v4]], %[[v17]]) {transpose_a = false, transpose_b = false} … 52 …// CHECK: %[[v28:.*]] = "tf.MatMul"(%[[v6]], %[[v19]]) {transpose_a = false, transpose_b = false} … 53 …// CHECK: %[[v29:.*]] = "tf.MatMul"(%[[v8]], %[[v21]]) {transpose_a = false, transpose_b = false} … 54 …// CHECK: %[[v30:.*]] = "tf.MatMul"(%[[v10]], %[[v23]]) {transpose_a = false, transpose_b = false}… 55 …// CHECK: %[[v31:.*]] = "tf.MatMul"(%[[v12]], %[[v25]]) {transpose_a = false, transpose_b = false}… 92 …// CHECK: %[[mm0:.*]] = "tf.MatMul"(%[[v1]], %[[v7]]) {transpose_a = false, transpose_b = false} :… 93 …// CHECK: %[[mm1:.*]] = "tf.MatMul"(%[[v3]], %[[v9]]) {transpose_a = false, transpose_b = false} :… 94 …// CHECK: %[[mm2:.*]] = "tf.MatMul"(%[[v5]], %[[v11]]) {transpose_a = false, transpose_b = false} … 108 …// CHECK: %[[v0:.*]] = "tf.MatMul"(%arg0, %arg1) {transpose_a = false, transpose_b = false} : (ten… [all …]
|
D | device_copy.mlir | 6 // CHECK: tf.MatMul 7 …%outputs, %control = tf_executor.island wraps "tf.MatMul"(%arg0, %arg1) {device = "", transpose_a … 18 // CHECK: tf.MatMul 19 …%outputs, %control = tf_executor.island wraps "tf.MatMul"(%arg0, %arg1) {device = "/device:GPU:0",…
|
D | contraction_fusion.mlir | 8 …%3 = "tf.MatMul"(%arg1, %arg2) {transpose_a = false, transpose_b = false} : (tensor<8x32xf32>, ten… 19 …%3 = "tf.MatMul"(%arg1, %arg2) {transpose_a = false, transpose_b = false} : (tensor<8x32xf32>, ten… 32 …%3 = "tf.MatMul"(%arg1, %arg2) {transpose_a = false, transpose_b = false} : (tensor<8x32xf32>, ten…
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
D | MatMul.pbtxt | 2 name: "MatMul" 45 name: "MatMul" 89 name: "MatMul" 133 name: "MatMul"
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
D | MatMul.pbtxt | 2 name: "MatMul" 45 name: "MatMul" 89 name: "MatMul" 133 name: "MatMul"
|
/external/tensorflow/tensorflow/core/api_def/java_api/ |
D | api_def_MatMul.pbtxt | 2 graph_op_name: "MatMul" 4 name: "linalg.MatMul"
|
/external/llvm-project/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/ |
D | saved_model.pbtxt | 3092 name: "MatMul/ReadVariableOp" 3117 original_node_names: "MatMul/ReadVariableOp" 3121 name: "MatMul" 3122 op: "MatMul" 3124 input: "MatMul/ReadVariableOp:value:0" 3147 original_node_names: "MatMul" 3179 input: "MatMul:product:0" 4499 name: "dense/MatMul/ReadVariableOp" 4524 original_node_names: "dense/MatMul/ReadVariableOp" 4528 name: "dense/MatMul" [all …]
|
/external/tensorflow/tensorflow/core/kernels/ |
D | immutable_constant_op_test.cc | 102 auto result = ops::MatMul(root, node1, node2); in TEST() 134 auto result = ops::MatMul(root, node1, node2); in TEST() 174 auto result = ops::MatMul(root, node1, node2, ops::MatMul::TransposeB(true)); in TEST()
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_MatMul.pbtxt | 2 graph_op_name: "MatMul" 22 *Note*: The default kernel implementation for MatMul on GPUs uses
|
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Scalar/ |
D | LowerMatrixIntrinsics.cpp | 703 void LowerMultiply(CallInst *MatMul) { in LowerMultiply() argument 704 IRBuilder<> Builder(MatMul); in LowerMultiply() 705 auto *EltType = cast<VectorType>(MatMul->getType())->getElementType(); in LowerMultiply() 706 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in LowerMultiply() 707 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in LowerMultiply() 710 getMatrix(MatMul->getArgOperand(0), LShape, Builder); in LowerMultiply() 712 getMatrix(MatMul->getArgOperand(1), RShape, Builder); in LowerMultiply() 728 bool AllowContract = AllowContractEnabled || (isa<FPMathOperator>(MatMul) && in LowerMultiply() 729 MatMul->hasAllowContract()); in LowerMultiply() 751 finalizeLowering(MatMul, Result, Builder); in LowerMultiply()
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | kernel_and_device_test.cc | 76 auto M = ops::MatMul(root, C, C); in BM_CreateGraph() 85 auto M = ops::MatMul(root, C, C); in BM_RunGraph() 101 auto M = ops::MatMul(root, C, C); in BM_CreateAndDestroySession()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | auto_mixed_precision_test.cc | 172 Output allow1 = ops::MatMul(s.WithOpName("allow1"), input, eye); in TestSimpleUnaryInferOp() 174 Output allow2 = ops::MatMul(s.WithOpName("allow2"), infer1, eye); in TestSimpleUnaryInferOp() 250 Output allow1 = ops::MatMul(s.WithOpName("allow1"), cst1, cst1); in TEST_F() 291 Output allow1 = ops::MatMul(s.WithOpName("allow1"), clr2, clr2); in TEST_F() 338 Output allow1 = ops::MatMul(s.WithOpName("allow1"), clr1, clr1); in TEST_F() 375 Output allow1 = ops::MatMul(s.WithOpName("allow1"), input, input); in TEST_F() 380 Output allow2 = ops::MatMul(s.WithOpName("allow2"), clr2, clr2); in TEST_F() 421 Output allow1 = ops::MatMul(s.WithOpName("allow1"), clr1, clr1); in TEST_F() 424 ops::MatMul(s.WithOpName("allow2").WithDevice( in TEST_F() 463 Output allow1 = ops::MatMul(s.WithOpName("allow1"), input, clr1); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/go/op/ |
D | scope_test.go | 119 square := MatMul(s.SubScope("square"), matrix, matrix) 121 cube := MatMul(s.SubScope("cube"), square, matrix) 177 output := MatMul(s,
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | runtime_single_threaded_matmul.cc | 33 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, in MatMul() function 76 MatMul<T, Eigen::Unaligned>(run_options_ptr, out, lhs, rhs, m, n, k, in SingleThreadedMatMulDispatch() 80 MatMul<T, Eigen::Aligned16>(run_options_ptr, out, lhs, rhs, m, n, k, in SingleThreadedMatMulDispatch()
|
D | runtime_matmul.cc | 37 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, in MatMul() function 83 MatMul<T, Eigen::Unaligned>(run_options_ptr, out, lhs, rhs, m, n, k, in MatMulDispatch() 88 MatMul<T, Eigen::Aligned16>(run_options_ptr, out, lhs, rhs, m, n, k, in MatMulDispatch()
|
/external/tensorflow/tensorflow/core/profiler/g3doc/ |
D | profile_model_architecture.md | 71 gradients/pool_logit/xw_plus_b/MatMul_grad/MatMul (163.84k/163.84k flops) 75 pool_logit/xw_plus_b/MatMul (163.84k/163.84k flops) 85 MatMul 491.52k float_ops (0.00%, 0.00%)
|