| /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()
|
| D | gradient_checker_test.cc | 32 using ops::MatMul;
132 auto z = MatMul(scope, x, y); in TEST()
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_MatMul.pbtxt | 2 graph_op_name: "MatMul"
4 name: "linalg.MatMul"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | immutable_constant_op_test.cc | 98 auto result = ops::MatMul(root, node1, node2); in TEST()
130 auto result = ops::MatMul(root, node1, node2); in TEST()
170 auto result = ops::MatMul(root, node1, node2, ops::MatMul::TransposeB(true)); in TEST()
|
| D | matmul_op.h | 43 void MatMul(const Device& d, Out out, In0 in0, In1 in1, in MatMul() function
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 69 auto matmul = ops::MatMul(s.WithOpName("matmul"), feed, const_1); in TEST()
72 ops::MatMul::Attrs matmul_attrs; in TEST()
74 auto incompatible_matmul = ops::MatMul(s.WithOpName("incompatible_matmul"), in TEST()
85 ops::MatMul(s.WithOpName("matmul_with_incompatible_input"), in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_stream_util_test.cc | 68 ops::MatMul(root, {}, {}); in TEST_F()
87 ops::MatMul(root, {}, {}); in TEST_F()
110 Output n = ops::MatMul(root, {}, {}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 37 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64 m, in MatMul() function
77 MatMul<T, Eigen::Unaligned>(run_options_ptr, out, lhs, rhs, m, n, k, in SingleThreadedMatMulDispatch()
84 MatMul<T, Eigen::Aligned16>(run_options_ptr, out, lhs, rhs, m, n, k, in SingleThreadedMatMulDispatch()
|
| D | runtime_matmul.cc | 40 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64 m, in MatMul() function
83 MatMul<T, Eigen::Unaligned>(run_options_ptr, out, lhs, rhs, m, n, k, in MatMulDispatch()
92 MatMul<T, Eigen::Aligned16>(run_options_ptr, out, lhs, rhs, m, n, k, in MatMulDispatch()
|
| /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/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 74 auto M = ops::MatMul(root, C, C); in BM_CreateGraph()
84 auto M = ops::MatMul(root, C, C); in BM_RunGraph()
102 auto M = ops::MatMul(root, C, C); in BM_CreateAndDestroySession()
|
| /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/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%)
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_MatMul.pbtxt | 2 graph_op_name: "MatMul"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_grad_test.cc | 248 Tensor MatMul(const Tensor& x, bool ax, const Tensor& y, bool ay) { in MatMul() function in tensorflow::__anon3cec93f10111::MathGradTest
1091 test::ExpectClose(dx, MatMul(dz, false, y, true)); in TEST_F()
1092 test::ExpectClose(dy, MatMul(x, true, dz, false)); in TEST_F()
1103 test::ExpectClose(dx, MatMul(dz, false, y, false)); in TEST_F()
1104 test::ExpectClose(dy, MatMul(dz, true, x, false)); in TEST_F()
1115 test::ExpectClose(dx, MatMul(y, false, dz, true)); in TEST_F()
1116 test::ExpectClose(dy, MatMul(x, false, dz, false)); in TEST_F()
1127 test::ExpectClose(dx, MatMul(y, true, dz, true)); in TEST_F()
1128 test::ExpectClose(dy, MatMul(dz, true, x, true)); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/python/testdata/ |
| D | permute.pbtxt | 73 op: "MatMul"
|
| /external/tensorflow/tensorflow/python/debug/examples/ |
| D | examples_test.sh | 82 run --node_name_filter hidden --op_type_filter MatMul
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model_test.cc | 45 auto m = ops::MatMul(root, placeholder, constant_tensor); in CreateTestGraph()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | analytical_cost_estimator_test.cc | 79 auto matmul = ops::MatMul(s.WithOpName("matmul"), flat, w2); in CreateMiniGraph()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass_test.cc | 949 Output matmul0 = ops::MatMul(root.WithOpName("MatMul0_dev0"), a, a); in TEST()
950 Output matmul1 = ops::MatMul(root.WithOpName("MatMul1_dev1"), b, b); in TEST()
953 ops::MatMul(root.WithOpName("MatMulCombined_dev1"), matmul0, matmul1); in TEST()
999 ops::MatMul(root.WithOpName("MatMulSource_dev0"), a, a); in TEST()
1001 Output matmul0 = ops::MatMul(root.WithOpName("MatMul0_dev0"), matmul_source, in TEST()
1003 Output matmul1 = ops::MatMul(root.WithOpName("MatMul1_dev1"), matmul_source, in TEST()
1201 Output y = ops::MatMul(root.WithOpName("test/y"), a, b); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding_test.cc | 89 auto m1 = ops::MatMul(s, a, b); in BuildSimpleGraph()
91 auto m2 = ops::MatMul(s.WithOpName("m2"), b, c); in BuildSimpleGraph()
728 auto result1 = ops::MatMul(root, a, b); in TEST_F()
729 auto result2 = ops::MatMul(root, result1, c); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | lstm_layer_inference.pbtxt | 2605 name: "lstm_0/MatMul"
2606 op: "MatMul"
2655 input: "lstm_0/MatMul"
3259 name: "lstm_1/MatMul"
3260 op: "MatMul"
3309 input: "lstm_1/MatMul"
3913 name: "lstm_2/MatMul"
3914 op: "MatMul"
3963 input: "lstm_2/MatMul"
4567 name: "lstm_3/MatMul"
[all …]
|
| /external/tensorflow/tensorflow/cc/profiler/ |
| D | profiler_test.cc | 42 auto y = ops::MatMul(root.WithOpName("y"), a, x); in CreateGraphDef()
|
| /external/tensorflow/tensorflow/contrib/util/ |
| D | convert_graphdef_memmapped_format_test.cc | 55 Output m = ops::MatMul(root, test_tensor1, test_tensor2); in TEST()
|