Searched defs:transpose_lhs (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_matmul_mkl.h | 49 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLMatMulF32()
58 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLMatMulF64()
67 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
76 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
|
| 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.cc | 41 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in MatMul()
77 int64 m, int64 n, int64 k, int32 transpose_lhs, in MatMulDispatch()
101 Eigen::half* rhs, int64 m, int64 n, int64 k, int32 transpose_lhs, in __xla_cpu_runtime_EigenMatMulF16()
109 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF32()
116 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF64()
|
| D | runtime_single_threaded_matmul.cc | 38 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in MatMul()
72 int32 transpose_lhs, int32 transpose_rhs) { in SingleThreadedMatMulDispatch()
94 Eigen::half* rhs, int64 m, int64 n, int64 k, int32 transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
104 int64 k, int32 transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
114 int64 k, int32 transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
|
| D | cpu_runtime_test.cc | 79 bool transpose_lhs, in EigenMatrixMultiply()
141 bool transpose_lhs = std::get<1>(info.param); in Name() local
154 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local
178 bool transpose_lhs = std::get<1>(info.param); in Name() local
191 bool transpose_lhs, in MKLMatrixMultiply()
223 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local
|
| D | runtime_matvec.h | 65 int32 transpose_lhs, int32 transpose_rhs) { in DispatchMatVec()
112 tensorflow::int32 transpose_lhs, in EigenMatVec()
|
| D | dot_op_emitter.cc | 663 bool transpose_lhs = mat_mult_dims.lhs_non_canonical; in EmitCallToRuntime() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1620 int64 k, int32 transpose_lhs, int32 transpose_rhs)>& impl_fn) { in MatmulArray2DImpl()
|
| D | algebraic_simplifier_test.cc | 4387 bool transpose_lhs, transpose_rhs; in TEST_P() local
|