| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 34 int32_t transpose_rhs) { in MatMul()
69 int32_t transpose_rhs) { in SingleThreadedMatMulDispatch()
88 int32_t transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
98 int32_t transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
109 int32_t transpose_rhs) { 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()
138 int32_t transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulS32()
|
| D | runtime_matmul.cc | 38 int32_t transpose_rhs) { in MatMul()
76 int32_t transpose_lhs, int32_t transpose_rhs) { in MatMul_Batch()
119 int32_t transpose_rhs) { in MatMulDispatch()
136 int32_t transpose_lhs, int32_t transpose_rhs) { in BatchMatMulDispatch()
154 int32_t transpose_rhs) { 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()
169 int32_t transpose_rhs) { 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()
193 int32_t transpose_rhs) { in __xla_cpu_runtime_EigenMatMulS32()
[all …]
|
| D | runtime_matmul_mkl.cc | 35 int32_t transpose_rhs) { 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()
92 int32_t transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF64()
109 int32_t transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
122 int32_t transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
|
| 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()
|
| D | cpu_runtime_test.cc | 81 bool transpose_rhs, in EigenMatrixMultiply()
143 bool transpose_rhs = std::get<2>(info.param); in Name() local
156 bool transpose_rhs = std::get<2>(GetParam()); in TEST_P() local
181 bool transpose_rhs = std::get<2>(info.param); in Name() local
194 bool transpose_rhs, in MKLMatrixMultiply()
226 bool transpose_rhs = std::get<2>(GetParam()); in TEST_P() local
|
| D | runtime_matmul_acl.cc | 39 int32_t transpose_rhs) { in MatMulF32()
185 int64_t n, int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_ACLMatMulF32()
197 int32_t transpose_rhs) { in __xla_cpu_runtime_ACLBatchMatMulF32()
|
| 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()
|
| D | dot_op_emitter.cc | 897 bool transpose_rhs = !mat_mult_dims.rhs_canonical; in EmitCallToRuntime() local
979 bool transpose_rhs = !mat_mult_dims.rhs_canonical; in EmitCallToBatchRuntime() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tridiagonal_solve_ops_test.py | 142 transpose_rhs=False): argument
160 transpose_rhs=False): argument
386 transpose_rhs=False, argument
|
| /external/ComputeLibrary/examples/gemm_tuner/ |
| D | cl_gemm_reshaped_rhs_only.cpp | 60 bool transpose_rhs{ true }; /**< Transpose rhs matrix */ member
127 …SimpleOption<size_t> *transpose_rhs; /**< Transpose rhs matrix option (1 enable; 0 disabl… member in __anon37653f440111::GemmConfigOptions
|
| D | cl_gemmlowp_reshaped.cpp | 64 bool transpose_rhs{ true }; /**< Transpose rhs matrix. */ member
142 …SimpleOption<size_t> *transpose_rhs; /**< Transpose rhs matrix option (1 enable; 0 disable). Also … member in __anon0359dd190111::GemmConfigOptions
|
| D | cl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint.cpp | 64 bool transpose_rhs{ true }; /**< Transpose rhs matrix */ member
127 SimpleOption<size_t> *transpose_rhs; /**< Transpose rhs matrix option (1 enable; 0 disable) */ member in __anon486f3f580111::GemmConfigOptions
|
| D | cl_gemm_reshaped.cpp | 64 bool transpose_rhs{ true }; /**< Transpose rhs matrix. */ member
146 …SimpleOption<size_t> *transpose_rhs; /**< Transpose rhs matrix option (1 enable; 0 disabl… member in __anon687f97770111::GemmConfigOptions
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | tridiagonal_solve_op_test.py | 74 transpose_rhs=False, argument
102 transpose_rhs=False, argument
389 transpose_rhs=False, argument
423 transpose_rhs=False, argument
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 448 transpose_rhs=False, argument
617 def _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs, argument
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | GEMMLowpFixture.h | 1249 … unsigned int k0, unsigned int h0, bool interleave_rhs, bool transpose_rhs, DataType data_type) in setup()
1392 …unsigned int k0, unsigned int h0, bool interleave_rhs, bool transpose_rhs, bool broadcast_bias, Da… in setup()
1600 … unsigned int k0, unsigned int h0, bool interleave_rhs, bool transpose_rhs, DataType data_type) in setup()
1756 … unsigned int k0, unsigned int h0, bool interleave_rhs, bool transpose_rhs, DataType data_type) in setup()
|
| D | GEMMFixture.h | 1412 …bool interleave_rhs, bool transpose_rhs, bool export_to_cl_image, DataType data_type, float alpha,… in setup()
1580 …bool interleave_rhs, bool transpose_rhs, bool export_to_cl_image, DataType data_type, float alpha,… in setup()
1815 …bool interleave_rhs, bool transpose_rhs, bool export_to_cl_image, bool has_pad_y, DataType data_ty… in setup()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1362 bool transpose_rhs = tfl_mm_op.adj_y(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 442 bool transpose_rhs, int64_t num_batch_dims, in BatchDot()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 5885 bool transpose_lhs, transpose_rhs; in TEST_P() local
|