Home
last modified time | relevance | path

Searched refs:output_num_cols (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc210 int64 output_num_cols = output_shape.dimensions(col_dim); in RunGemm() local
254 std::swap(output_num_cols, output_num_rows); in RunGemm()
258 output_num_rows, output_num_cols}; in RunGemm()
/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc623 const int output_num_cols = output->dimension(1); in GEPP() local
631 out_ptrs[j] = out_start + output_num_cols * j; in GEPP()