| /external/tensorflow/tensorflow/cc/framework/ |
| gradients_test.cc | 34 using ops::MatMul;
64 // dy| dx| (MatMul Gradient Graph)
75 // | z| | (MatMul Forward Graph)
93 auto z = MatMul(scope, x, y);
100 auto dx = MatMul(scope, dz, y, MatMul::TransposeB(true));
101 auto dy = MatMul(scope, x, dz, MatMul::TransposeA(true));
119 auto z = MatMul(scope, x, y);
128 auto dx = MatMul(scope, dz, y, MatMul::TransposeB(true))
[all...] |
| cc_ops_test.cc | 30 auto m = MatMul(cop_scopes.child, x, w);
50 MatMul m(root, c, {{41}, {1}});
59 auto m = MatMul(root, {{1}, {1}}, {{41}, {1}}, MatMul::TransposeA(true));
104 auto y = MatMul(root.WithOpName("y"), a, x);
|
| gradient_checker_test.cc | 31 using ops::MatMul;
116 auto z = MatMul(scope, x, y);
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| evaluator.py | 43 "hidden_0/MatMul:0"
45 "/job:worker/replica:0/task:1/gpu:0:hidden_0/MatMul:0"
47 "hidden_0/MatMul:0:DebugNumericSummary"
49 "/job:worker/replica:0/task:1/gpu:0:hidden_0/MatMul:0:DebugNumericSummary"
51 "/job:worker/replica:0/task:1/gpu:0:hidden_0/MatMul:0:DebugNumericSummary[1]"
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_matmul.cc | 31 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64 m,
76 MatMul<float>(run_options_ptr, out, lhs, rhs, m, n, k, transpose_lhs,
89 MatMul<double>(run_options_ptr, out, lhs, rhs, m, n, k, transpose_lhs,
|
| runtime_single_threaded_matmul.cc | 28 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64 m,
68 MatMul<float>(run_options_ptr, out, lhs, rhs, m, n, k, transpose_lhs,
79 MatMul<double>(run_options_ptr, out, lhs, rhs, m, n, k, transpose_lhs,
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fake_quantize_training_test.cc | 50 Output matmul = MatMul(root.WithOpName("matmul"), a_const, b_const); local
|
| remove_ema_test.cc | 55 Output matmul = MatMul(root.WithOpName("matmul"), a_const, b_const); local
72 TF_ASSERT_OK(session->Run({}, {"matmul"}, {}, &outputs));
|
| fold_batch_norms_test.cc | 165 MatMul(root.WithOpName("matmul_op"), input_op, weights_op);
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_stream_util_test.cc | 68 ops::MatMul(root, {}, {});
87 ops::MatMul(root, {}, {});
110 Output n = ops::MatMul(root, {}, {});
|
| /external/tensorflow/tensorflow/python/debug/examples/ |
| examples_test.sh | 74 run --node_name_filter hidden --op_type_filter MatMul
|
| /external/tensorflow/tensorflow/core/kernels/ |
| immutable_constant_op_test.cc | 97 auto result = ops::MatMul(root, node1, node2);
129 auto result = ops::MatMul(root, node1, node2);
169 auto result = ops::MatMul(root, node1, node2, ops::MatMul::TransposeB(true));
|
| matmul_op.h | 27 // Helpers to define tensor<T> needed by MatMul op.
39 void MatMul(const Device& d, Out out, In0 in0, In1 in1,
58 // Encapsulate all the shape information that is used in matmul operations.
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| benchmark_model_test.cc | 48 auto m = ops::MatMul(root, placeholder, constant_tensor);
|
| /external/tensorflow/tensorflow/python/tools/ |
| print_selective_registration_header_test.py | 43 op: "MatMul"
52 op: "MatMul"
99 ('MatMul', 'MatMulOp<CPUDevice, double, false >'), #
100 ('MatMul', 'MatMulOp<CPUDevice, float, false >'), #
115 ('MatMul', 'MatMulOp<CPUDevice, double, false >'), #
116 ('MatMul', 'MatMulOp<CPUDevice, float, false >'), #
|
| /external/tensorflow/tensorflow/c/eager/ |
| runtime_test.cc | 61 s = AttrTypeMapForOp("MatMul", &m);
87 NodeDef ndef(AttrBuilder("MatMul")
113 auto M = ops::MatMul(root, C, C);
123 auto M = ops::MatMul(root, C, C);
141 auto M = ops::MatMul(root, C, C);
151 NodeDef ndef(AttrBuilder("MatMul")
174 NodeDef ndef(AttrBuilder("MatMul")
|
| /external/tensorflow/tensorflow/python/ops/ |
| matmul_benchmark_test.py | 99 node { name: "MatMul" op: "MatMul" input: "Variable/read" input: "Variable_1/read" device: \"""" + dev + """\" }
100 node { name: "group_deps" op: "NoOp" input: "^MatMul" device: \"""" + dev + """\" }
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| analytical_cost_estimator_test.cc | 80 auto matmul = ops::MatMul(s.WithOpName("matmul"), flat, w2); local
81 auto logits = ops::Add(s.WithOpName("logits"), matmul, b2);
|
| /external/tensorflow/tensorflow/core/profiler/g3doc/ |
| profile_time.md | 121 MatMul 618.97ms (63.56%, 16.51%), 3450
137 MatMul 618.97ms (63.56%, 16.51%), |/job:worker/replica:0/task:0/cpu:0|/job:worker/replica:0/task:0/device:GPU:0|/job:worker/replica:0/task:0/device:GPU:1|/job:worker/replica:0/task:0/device:GPU:2|/job:worker/replica:0/task:0/device:GPU:3
|
| /external/tensorflow/tensorflow/cc/profiler/ |
| profiler_test.cc | 42 auto y = ops::MatMul(root.WithOpName("y"), a, x);
137 const GraphNodeProto* matmul = ExtractNode(ret, "y"); local
138 EXPECT_TRUE(matmul);
139 EXPECT_GT(matmul->exec_micros(), 0);
141 EXPECT_GT(matmul->accelerator_exec_micros(), 0);
143 EXPECT_EQ(matmul->accelerator_exec_micros(), 0);
164 const MultiGraphNodeProto* matmul2 = ExtractNode(ret3, "MatMul");
|
| /external/tensorflow/tensorflow/contrib/util/ |
| convert_graphdef_memmapped_format_test.cc | 55 Output m = ops::MatMul(root, test_tensor1, test_tensor2);
|
| /external/tensorflow/tensorflow/core/ops/ |
| math_grad_test.cc | 231 Tensor MatMul(const Tensor& x, bool ax, const Tensor& y, bool ay) {
232 return MatMulCommon("MatMul", "transpose_a", "transpose_b", x, ax, y, ay);
301 return MatMulGradCommon("MatMul", "transpose_a", "transpose_b", x, ax, y,
[all...] |
| /external/tensorflow/tensorflow/cc/tutorials/ |
| example_trainer.cc | 63 auto y = MatMul(root.WithOpName("y"), a, x);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| constant_folding_test.cc | 94 auto m1 = ops::MatMul(s, a, b);
96 auto m2 = ops::MatMul(s.WithOpName("m2"), b, c);
690 auto result1 = ops::MatMul(root, a, b);
691 auto result2 = ops::MatMul(root, result1, c);
|
| /external/tensorflow/tensorflow/go/op/ |
| scope_test.go | 162 output := MatMul(s,
|
