Home
last modified time | relevance | path

Searched defs:transpose_rhs (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_matmul.cc35 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in MatMul()
71 tensorflow::int32 transpose_rhs) { in SingleThreadedMatMulDispatch()
91 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
100 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
109 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
119 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulC64()
129 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulC128()
139 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulS32()
Druntime_matmul.cc39 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in MatMul()
78 tensorflow::int32 transpose_rhs) { in MatMulDispatch()
98 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF16()
106 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF32()
114 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF64()
123 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulC64()
132 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulC128()
141 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulS32()
Druntime_matmul_mkl.h50 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32()
59 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF64()
68 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
77 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
Druntime_matmul_mkl.cc39 int32 transpose_rhs) { in MatMulF32()
61 int32 transpose_rhs) { in MatMulF64()
80 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32()
95 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF64()
112 int32 transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
125 int32 transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
Dcpu_runtime_test.cc79 bool transpose_rhs, in EigenMatrixMultiply()
140 bool transpose_rhs = std::get<2>(info.param); in Name() local
153 bool transpose_rhs = std::get<2>(GetParam()); in TEST_P() local
177 bool transpose_rhs = std::get<2>(info.param); in Name() local
190 bool transpose_rhs, in MKLMatrixMultiply()
222 bool transpose_rhs = std::get<2>(GetParam()); in TEST_P() local
Ddot_op_emitter.cc830 bool transpose_rhs = !mat_mult_dims.rhs_canonical; in EmitCallToRuntime() local
/external/tensorflow/tensorflow/compiler/tests/
Dtridiagonal_solve_ops_test.py146 transpose_rhs=False): argument
164 transpose_rhs=False): argument
390 transpose_rhs=False, argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dtridiagonal_solve_op_test.py78 transpose_rhs=False, argument
101 transpose_rhs=False, argument
396 transpose_rhs=False, argument
430 transpose_rhs=False, argument
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py451 transpose_rhs=False, argument
600 def _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs, argument
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc540 bool transpose_rhs, int64_t num_batch_dims, in BatchDot()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc5244 bool transpose_lhs, transpose_rhs; in TEST_P() local