Home
last modified time | relevance | path

Searched defs:lhs_rows (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_matmul.cc39 int64 lhs_rows = m; in MatMul() local
Druntime_matmul.cc45 int64 lhs_rows = m; in MatMul() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc67 const int64 lhs_rows = lhs_shape.dimensions(num_batch_dims + 0); in DecomposeBatchDot() local
Dalgebraic_simplifier_test.cc4640 int64 lhs_rows = (spec.lcd == 0) ? (spec.k + k_increase) : spec.m; in TEST_P() local
4710 int64 lhs_rows = (spec.lcd == 0) ? spec.k : spec.m; in TEST_P() local
/external/tensorflow/tensorflow/core/kernels/
Dmkl_batch_matmul_op.cc78 auto lhs_rows = lhs.dim_size(ndims - 2); in Compute() local