| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_matmul.cc | 37 int64_t n, int64_t k, int32_t transpose_lhs, in MatMul()
76 int32_t transpose_lhs, int32_t transpose_rhs) { in MatMul_Batch()
118 int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in MatMulDispatch()
136 int32_t transpose_lhs, int32_t transpose_rhs) { in BatchMatMulDispatch()
153 Eigen::half* rhs, int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenMatMulF16()
161 int64_t n, int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF32()
168 int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenMatMulF64()
177 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenMatMulC64()
185 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenMatMulC128()
192 int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenMatMulS32()
[all …]
|
| H A D | runtime_single_threaded_matmul.cc | 33 int64_t n, int64_t k, int32_t transpose_lhs, in MatMul()
68 int32_t transpose_lhs, in SingleThreadedMatMulDispatch()
87 Eigen::half* rhs, int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
97 int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
108 int32_t transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
118 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulC64()
127 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulC128()
137 int32_t transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulS32()
|
| H A D | runtime_matmul_mkl.cc | 34 int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in MatMulF32()
57 int32_t transpose_lhs, int32_t transpose_rhs) { in MatMulF64()
76 int64_t n, int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32()
91 int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_MKLMatMulF64()
108 int32_t transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
121 int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
|
| H A D | cpu_runtime_test.cc | 80 bool transpose_lhs, in EigenMatrixMultiply()
142 bool transpose_lhs = std::get<1>(info.param); in Name() local
155 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local
180 bool transpose_lhs = std::get<1>(info.param); in Name() local
193 bool transpose_lhs, in MKLMatrixMultiply()
225 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local
|
| H A D | runtime_matmul_mkl.h | 45 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32()
53 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF64()
61 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
69 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
|
| H A D | runtime_matmul_acl.cc | 38 int64_t batch_size, int32_t transpose_lhs, in MatMulF32()
185 int64_t n, int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_ACLMatMulF32()
196 int64_t n, int64_t k, int64_t batch_size, int32_t transpose_lhs, in __xla_cpu_runtime_ACLBatchMatMulF32()
|
| H A D | runtime_matmul_acl.h | 69 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_ACLMatMulF32()
79 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_ACLBatchMatMulF32()
|
| H A D | dot_op_emitter.cc | 896 bool transpose_lhs = !mat_mult_dims.lhs_canonical; in EmitCallToRuntime() local
978 bool transpose_lhs = !mat_mult_dims.lhs_canonical; in EmitCallToBatchRuntime() local
|
| /aosp_15_r20/external/ComputeLibrary/examples/gemm_tuner/ |
| H A D | cl_gemmlowp_reshaped.cpp | 62 bool transpose_lhs{ true }; /**< Transpose lhs matrix. */ member
|
| H A D | cl_gemm_reshaped.cpp | 62 bool transpose_lhs{ true }; /**< Transpose lhs matrix. */ member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_tfl.cc | 1361 bool transpose_lhs = tfl_mm_op.adj_x(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_evaluator.cc | 3882 int32_t transpose_lhs, int32_t transpose_rhs)>& in MatmulArray2DImpl()
|
| H A D | algebraic_simplifier_test.cc | 5885 bool transpose_lhs, transpose_rhs; in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 441 Value BatchDot(Location loc, Value lhs, bool transpose_lhs, Value rhs, in BatchDot()
|