Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_matmul.cc46 int64 rhs_cols = n; in MatMul() local
Druntime_matmul.cc52 int64 rhs_cols = n; in MatMul() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc79 const int64 rhs_cols = rhs_shape.dimensions(num_batch_dims + 1); in DecomposeBatchDot() local
Dalgebraic_simplifier_test.cc4663 int64 rhs_cols = (spec.rcd == 0) ? spec.n : spec.k; in TEST_P() local
4722 int64 rhs_cols = (spec.rcd == 0) ? spec.n : (spec.k + k_increase); in TEST_P() local
/external/tensorflow/tensorflow/core/kernels/
Dmkl_batch_matmul_op.cc81 auto rhs_cols = rhs.dim_size(ndims - 1); in Compute() local