Home
last modified time | relevance | path

Searched refs:MatMul (Results 1 – 25 of 116) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/
Dback2back_fake_quant.pbtxt31 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 …]
Dcontrol_flow_v1.pbtxt118 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"
Dunroll_batch_matmul.pbtxt1 …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/
Dgradients_test.cc34 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 …]
Dcc_ops_test.cc30 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/
Dopens2s_gnmt_mixed_precision.golden_summary119 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/
DLowerMatrixIntrinsics.cpp1091 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/
Dmnist_gradients_testutil.cc78 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/
Dmath_grad.cc28 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/
Dunroll-batch-matmul.mlir50 …// 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 …]
Ddevice_copy.mlir6 // 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",…
Dcontraction_fusion.mlir8 …%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/
DMatMul.pbtxt2 name: "MatMul"
45 name: "MatMul"
89 name: "MatMul"
133 name: "MatMul"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DMatMul.pbtxt2 name: "MatMul"
45 name: "MatMul"
89 name: "MatMul"
133 name: "MatMul"
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_MatMul.pbtxt2 graph_op_name: "MatMul"
4 name: "linalg.MatMul"
/external/llvm-project/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/
Dsaved_model.pbtxt3092 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/
Dimmutable_constant_op_test.cc102 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/
Dapi_def_MatMul.pbtxt2 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/
DLowerMatrixIntrinsics.cpp703 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/
Dkernel_and_device_test.cc76 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/
Dauto_mixed_precision_test.cc172 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/
Dscope_test.go119 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/
Druntime_single_threaded_matmul.cc33 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()
Druntime_matmul.cc37 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/
Dprofile_model_architecture.md71 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%)

12345