| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| 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,
|
| 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,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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/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/c/ |
| c_api_test.cc | 1570 TF_Operation* matmul = MatMul(graph_, s_, const0, const1, "MatMul"); local
1610 TF_Operation* matmul = local
[all...] |
| /external/deqp/external/vulkancts/modules/vulkan/shaderexecutor/ |
| vktShaderBuiltinPrecisionTests.cpp | [all...] |
| /external/deqp/modules/glshared/ |
| glsBuiltinPrecisionTests.cpp | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
