Home
last modified time | relevance | path

Searched defs:rhs_rows (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_matmul.cc48 int64_t rhs_rows = k; in MatMul() local
86 int64_t rhs_rows = k; in MatMul_Batch() local
H A Druntime_single_threaded_matmul.cc41 int64_t rhs_rows = k; in MatMul() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_batch_matmul_op.cc105 auto rhs_rows = rhs.dim_size(ndims_rhs - 2); in Compute() local
H A Dmkl_einsum_op.cc80 auto rhs_rows = rhs.dim_size(ndims_rhs - 2); in MKLContractOperands() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dcpu_backend_gemm_test.cc524 void TestMaybeValidGemm(int lhs_rows, int lhs_cols, int rhs_rows, int rhs_cols, in TestMaybeValidGemm()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier_test.cc6319 int64_t rhs_rows = (spec.rcd == 0) ? spec.k : spec.n; in TEST_P() local
6378 int64_t rhs_rows = (spec.rcd == 0) ? (spec.k + k_increase) : spec.n; in TEST_P() local