Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.h92 StatusOr<se::dnn::DataType> GetDNNDataTypeFromPrimitiveType(PrimitiveType type);
Dstream_executor_util.cc337 StatusOr<se::dnn::DataType> GetDNNDataTypeFromPrimitiveType( in GetDNNDataTypeFromPrimitiveType() function
Dgpu_conv_algorithm_picker.cc137 GetDNNDataTypeFromPrimitiveType(config.output_type)); in GetMIOpenAlgorithms()