Home
last modified time | relevance | path

Searched refs:GetDNNDataTypeFromPrimitiveType (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_runner.cc87 GetDNNDataTypeFromPrimitiveType(params.config->input_type)); in RunGpuConvUnfused()
91 GetDNNDataTypeFromPrimitiveType(params.config->output_type)); in RunGpuConvUnfused()
151 GetDNNDataTypeFromPrimitiveType(params.config->input_type)); in RunGpuConvForwardActivation()
155 GetDNNDataTypeFromPrimitiveType(params.config->output_type)); in RunGpuConvForwardActivation()
H A Dstream_executor_util.h101 StatusOr<se::dnn::DataType> GetDNNDataTypeFromPrimitiveType(PrimitiveType type);
H A Dgpu_conv_algorithm_picker.cc116 GetDNNDataTypeFromPrimitiveType(config.input_type)); in GetAlgorithms()
119 GetDNNDataTypeFromPrimitiveType(config.output_type)); in GetAlgorithms()
197 GetDNNDataTypeFromPrimitiveType(config.output_type)); in GetMIOpenAlgorithms()
H A Dstream_executor_util.cc495 StatusOr<se::dnn::DataType> GetDNNDataTypeFromPrimitiveType( in GetDNNDataTypeFromPrimitiveType() function