Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_batch_matmul_op.cc107 auto lhs_cols = lhs.dim_size(ndims_lhs - 1); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_matmul.cc37 tensorflow::int64 lhs_cols = k; in MatMul() local
Druntime_matmul.cc44 tensorflow::int64 lhs_cols = k; in MatMul() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc5601 int64 lhs_cols = (spec.lcd == 0) ? spec.m : (spec.k + k_increase); in TEST_P() local
5671 int64 lhs_cols = (spec.lcd == 0) ? spec.m : spec.k; in TEST_P() local