Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_matmul.cc38 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 …]
H A Druntime_single_threaded_matmul.cc34 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()
H A Druntime_matmul_mkl.cc35 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()
H A Dcpu_runtime_test.cc81 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
H A Druntime_matmul_mkl.h45 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 Druntime_matmul_acl.cc39 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()
H A Druntime_matmul_acl.h69 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 Ddot_op_emitter.cc897 bool transpose_rhs = !mat_mult_dims.rhs_canonical; in EmitCallToRuntime() local
979 bool transpose_rhs = !mat_mult_dims.rhs_canonical; in EmitCallToBatchRuntime() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dtridiagonal_solve_op_test.py74 transpose_rhs=False, argument
102 transpose_rhs=False, argument
389 transpose_rhs=False, argument
423 transpose_rhs=False, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dtridiagonal_solve_ops_test.py142 transpose_rhs=False): argument
160 transpose_rhs=False): argument
386 transpose_rhs=False, argument
/aosp_15_r20/external/ComputeLibrary/examples/gemm_tuner/
H A Dcl_gemm_reshaped_rhs_only.cpp60 bool transpose_rhs{ true }; /**< Transpose rhs matrix */ member
127 …SimpleOption<size_t> *transpose_rhs; /**< Transpose rhs matrix option (1 enable; 0 disabl… member in __anon0efe27150111::GemmConfigOptions
H A Dcl_gemmlowp_reshaped.cpp64 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 __anon0f4d5e4a0111::GemmConfigOptions
H A Dcl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint.cpp64 bool transpose_rhs{ true }; /**< Transpose rhs matrix */ member
127 SimpleOption<size_t> *transpose_rhs; /**< Transpose rhs matrix option (1 enable; 0 disable) */ member in __anoncf1563690111::GemmConfigOptions
H A Dcl_gemm_reshaped.cpp64 bool transpose_rhs{ true }; /**< Transpose rhs matrix. */ member
146 …SimpleOption<size_t> *transpose_rhs; /**< Transpose rhs matrix option (1 enable; 0 disabl… member in __anona62828280111::GemmConfigOptions
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DGEMMLowpFixture.h1249 … 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()
H A DGEMMFixture.h1412 …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()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/
H A Dlinalg_impl.py448 transpose_rhs=False, argument
617 def _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tfl.cc1362 bool transpose_rhs = tfl_mm_op.adj_y(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc442 bool transpose_rhs, int64_t num_batch_dims, in BatchDot()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier_test.cc5885 bool transpose_lhs, transpose_rhs; in TEST_P() local