Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_batch_matmul_op.cc108 auto rhs_rows = rhs.dim_size(ndims_rhs - 2); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_matmul.cc42 tensorflow::int64 rhs_rows = k; in MatMul() local
Druntime_matmul.cc49 tensorflow::int64 rhs_rows = k; in MatMul() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc5622 int64 rhs_rows = (spec.rcd == 0) ? spec.k : spec.n; in TEST_P() local
5681 int64 rhs_rows = (spec.rcd == 0) ? (spec.k + k_increase) : spec.n; in TEST_P() local