/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_conv_runner.cc | 181 BatchDescriptor input_descriptor(effective_num_dimensions); in RunCudnnConvImpl() local 182 input_descriptor.set_layout(input_dl) in RunCudnnConvImpl() 188 input_descriptor.set_spatial_dim( in RunCudnnConvImpl() 231 input_descriptor.set_spatial_dim(static_cast<DimIndex>(0), 1); in RunCudnnConvImpl() 246 input_descriptor, input_buf, filter_descriptor, filter_buf, in RunCudnnConvImpl() 258 convolution_descriptor, input_descriptor, &input_buf, in RunCudnnConvImpl() 268 input_descriptor, input_buf, output_descriptor, output_buf, in RunCudnnConvImpl() 299 input_descriptor, input_buf, params.conv_result_scale, in RunCudnnConvImpl()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.cc | 320 miopenTensorDescriptor_t input_descriptor) { in FindOrCreate() argument 331 input_descriptor); in FindOrCreate() 914 const miopenTensorDescriptor_t input_descriptor) in ScopedFusionPlanBase() argument 934 miopenStatus_t Execute(miopenTensorDescriptor_t input_descriptor, in Execute() argument 939 miopen_handle_, fusion_plan_, input_descriptor, input_data, in Execute() 1118 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanConvolutionBiasActivation() argument 1124 input_descriptor) { in ScopedFusionPlanConvolutionBiasActivation() 1125 uint64 hash = GetFusionOpHashValue(miopen_handle, input_descriptor, in ScopedFusionPlanConvolutionBiasActivation() 1130 hash, &fusion_plan_, miopenVerticalFusion, input_descriptor); in ScopedFusionPlanConvolutionBiasActivation() 1198 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue() argument [all …]
|
D | rocm_dnn.h | 255 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, 325 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized() argument 338 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized() argument 364 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 370 const dnn::BatchDescriptor& input_descriptor, 376 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 657 const dnn::BatchDescriptor& input_descriptor, 760 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 264 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, 334 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized() argument 347 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized() argument 373 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 379 const dnn::BatchDescriptor& input_descriptor, 385 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 600 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 653 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
|
D | cuda_dnn.cc | 2627 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution() argument 2636 input_descriptor, in DoPrepareForConvolution() 2637 ToCudnnDataType(element_type, input_descriptor.layout())); in DoPrepareForConvolution() 2685 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve() argument 2693 CudnnTensorDescriptor input_nd(input_descriptor, cudnn_type); in DoConvolve() 2732 if (input_descriptor.ndims() < 3) { in DoConvolve() 2744 SE_RETURN_IF_ERROR(check_sizes(input_descriptor.feature_map_count(), in DoConvolve() 2746 SE_RETURN_IF_ERROR(check_sizes(input_descriptor.count(), in DoConvolve() 2747 input_descriptor.feature_map_count())); in DoConvolve() 2748 SE_RETURN_IF_ERROR(check_sizes(input_descriptor.count(), in DoConvolve() [all …]
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 1242 const BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, 1250 bool DoConvolve(Stream* stream, const dnn::BatchDescriptor& input_descriptor, 1262 stream, input_descriptor, input_data, filter_descriptor, 1284 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 1295 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 1313 Stream* stream, const BatchDescriptor& input_descriptor, 1347 const dnn::BatchDescriptor& input_descriptor, 1354 ToDataType<ElementType>::value, stream, input_descriptor, 1389 Stream* stream, const BatchDescriptor& input_descriptor, 1401 ToDataType<ElementType>::value, stream, input_descriptor, [all …]
|
D | stream.cc | 594 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm() argument 603 VLOG_CALL(PARAM(input_descriptor), PARAM(input_data), in ThenConvolveWithAlgorithm() 614 dnn::ConvolutionKind::FORWARD, this, input_descriptor, in ThenConvolveWithAlgorithm() 621 this, input_descriptor, input_data, filter_descriptor, filter_data, in ThenConvolveWithAlgorithm() 636 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm() argument 645 VLOG_CALL(PARAM(input_descriptor), PARAM(input_data), in ThenConvolveWithAlgorithm() 656 dnn::ConvolutionKind::FORWARD, this, input_descriptor, in ThenConvolveWithAlgorithm() 663 this, input_descriptor, input_data, filter_descriptor, filter_data, in ThenConvolveWithAlgorithm() 678 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm() argument 687 VLOG_CALL(PARAM(input_descriptor), PARAM(input_data), in ThenConvolveWithAlgorithm() [all …]
|
D | stream.h | 276 Stream &ThenConvolve(const dnn::BatchDescriptor &input_descriptor, 285 const dnn::BatchDescriptor &input_descriptor, 295 const dnn::BatchDescriptor &input_descriptor, 305 const dnn::BatchDescriptor &input_descriptor, 316 const dnn::BatchDescriptor &input_descriptor, 327 const dnn::BatchDescriptor &input_descriptor, 395 const dnn::BatchDescriptor &input_descriptor, 410 const dnn::BatchDescriptor &input_descriptor, 422 const dnn::BatchDescriptor &input_descriptor, 434 const dnn::BatchDescriptor &input_descriptor, [all …]
|