| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 34 int64_t n, int64_t k, int32_t transpose_lhs, in MatMul()
69 int32_t transpose_lhs, in SingleThreadedMatMulDispatch()
88 Eigen::half* rhs, int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
98 int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
109 int32_t transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
119 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulC64()
128 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulC128()
137 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulS32()
|
| D | runtime_matmul.cc | 38 int64_t n, int64_t k, int32_t transpose_lhs, in MatMul()
76 int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in MatMulDispatch()
95 Eigen::half* rhs, int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenMatMulF16()
103 int64_t n, int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF32()
110 int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenMatMulF64()
119 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenMatMulC64()
127 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenMatMulC128()
135 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenMatMulS32()
|
| D | runtime_matmul_mkl.cc | 38 int64 m, int64 n, int64 k, int32 transpose_lhs, in MatMulF32()
60 double* rhs, int64 m, int64 n, int64 k, int32 transpose_lhs, 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()
111 int32 transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
124 int64 k, int32 transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
|
| D | runtime_matmul_mkl.h | 49 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32()
57 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF64()
65 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
73 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
|
| D | cpu_runtime_test.cc | 78 bool transpose_lhs, in EigenMatrixMultiply()
139 bool transpose_lhs = std::get<1>(info.param); in Name() local
152 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local
176 bool transpose_lhs = std::get<1>(info.param); in Name() local
189 bool transpose_lhs, in MKLMatrixMultiply()
221 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local
|
| D | dot_op_emitter.cc | 834 bool transpose_lhs = !mat_mult_dims.lhs_canonical; in EmitCallToRuntime() local
|
| /external/ComputeLibrary/examples/gemm_tuner/ |
| D | cl_gemm_reshaped.cpp | 61 bool transpose_lhs{ true }; /**< Transpose lhs matrix. */ member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 2624 int32_t transpose_lhs, int32_t transpose_rhs)>& in MatmulArray2DImpl()
|
| D | algebraic_simplifier_test.cc | 5277 bool transpose_lhs, transpose_rhs; in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 489 Value BatchDot(Location loc, Value lhs, bool transpose_lhs, Value rhs, in BatchDot()
|