Home
last modified time | relevance | path

Searched refs:scratch_size_override (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h42 absl::optional<size_t> scratch_size_override; member
Dgpu_conv_runner.cc231 if (options.scratch_size_override.has_value()) { in RunGpuConvImpl()
232 algorithm.set_scratch_size(*options.scratch_size_override); in RunGpuConvImpl()
Dgpu_conv_algorithm_picker.cc726 options.scratch_size_override = miopen_alg.scratch_size(); in PickBestAlgorithmNoCacheRocm()