Searched refs:scratch_bytes (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_convolution_algorithm_picker.cc298 int64 scratch_bytes; local
300 std::tie(algorithm, tensor_ops_enabled, scratch_bytes) = *alg_scratch_and_tc;
303 << NumBytesToString(scratch_bytes)
312 ShapeUtil::MakeShape(U8, {scratch_bytes})});

Completed in 87 milliseconds