Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc209 int64 output_num_rows = output_shape.dimensions(row_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/python/debug/cli/
Dcurses_ui.py93 output_num_rows): argument
117 self._output_num_rows = output_num_rows