Home
last modified time | relevance | path

Searched refs:GetConvAccumulatorType (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2579 dnn::DataType GetConvAccumulatorType(dnn::DataType data_type) { in GetConvAccumulatorType() function
2646 ToCudnnDataType(GetConvAccumulatorType(element_type))); in DoPrepareForConvolution()
2696 auto accumulator_type = GetConvAccumulatorType(element_type); in DoConvolve()
3362 GetConvAccumulatorType(dnn::DataType::kDouble), scratch_allocator, in DoFusedConvolve()
3386 GetConvAccumulatorType(dnn::DataType::kFloat), scratch_allocator, in DoFusedConvolve()
3411 GetConvAccumulatorType(dnn::DataType::kHalf), scratch_allocator, in DoFusedConvolve()
3443 GetConvAccumulatorType(dnn::DataType::kInt8), scratch_allocator, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1607 dnn::DataType GetConvAccumulatorType(dnn::DataType data_type) { in GetConvAccumulatorType() function
2612 ToMIOpenDataType(GetConvAccumulatorType(element_type))}; in DoPrepareForConvolution()