Home
last modified time | relevance | path

Searched defs:use_fallback (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream_executor_pimpl.cc300 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetConvolveRunners()
323 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetFusedConvolveRunners()
H A Ddnn.cc151 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetFusedConvolveRunners()
/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dstructure.py426 def type_spec_from_value(element, use_fallback=True): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_algorithm_picker.cc111 bool use_fallback) { in GetAlgorithms()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc4594 absl::Span<const int64_t> input_uids, bool use_fallback, in CreateOpRunners()
4719 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetConvolveRunners()
5146 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetFusedConvolveRunners()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc3153 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetConvolveRunners()