/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_conv_runner.cc | 89 params.config.input_descriptor, input_buf, in RunGpuConvForward() 129 params.config.input_descriptor, input_buf, in RunGpuConvForwardActivation() 169 params.config.input_descriptor, &input_buf, scratch_allocator, in RunGpuConvInternalImpl() 179 params.config.input_descriptor, input_buf, in RunGpuConvInternalImpl() 374 BatchDescriptor& input_descriptor = config.input_descriptor; in GetGpuConvConfig() local 375 input_descriptor = BatchDescriptor(effective_num_dimensions); in GetGpuConvConfig() 376 input_descriptor.set_layout(input_dl) in GetGpuConvConfig() 382 input_descriptor.set_spatial_dim( in GetGpuConvConfig() 430 input_descriptor.set_spatial_dim(static_cast<DimIndex>(dim), 1); in GetGpuConvConfig()
|
D | gpu_conv_runner.h | 70 se::dnn::BatchDescriptor input_descriptor; member
|
D | gpu_conv_algorithm_picker.cc | 143 kind, dtype, stream, params.config.input_descriptor, params.input_buf, in GetMIOpenAlgorithms()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.cc | 435 miopenTensorDescriptor_t input_descriptor) { in FindOrCreate() argument 446 input_descriptor); in FindOrCreate() 1031 const miopenTensorDescriptor_t input_descriptor) in ScopedFusionPlanBase() argument 1051 miopenStatus_t Execute(miopenTensorDescriptor_t input_descriptor, in Execute() argument 1056 miopen_handle_, fusion_plan_, input_descriptor, input_data, in Execute() 1231 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanConvolutionBiasActivation() argument 1237 input_descriptor) { in ScopedFusionPlanConvolutionBiasActivation() 1238 uint64 hash = GetFusionOpHashValue(miopen_handle, input_descriptor, in ScopedFusionPlanConvolutionBiasActivation() 1243 hash, &fusion_plan_, miopenVerticalFusion, input_descriptor); in ScopedFusionPlanConvolutionBiasActivation() 1311 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue() argument [all …]
|
D | rocm_dnn.h | 236 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, 309 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, 377 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized() argument 390 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized() argument 416 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 422 const dnn::BatchDescriptor& input_descriptor, 428 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 732 const dnn::BatchDescriptor& input_descriptor, 835 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, 866 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, [all …]
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 271 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, 353 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized() argument 366 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized() argument 392 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 398 const dnn::BatchDescriptor& input_descriptor, 404 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 634 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 696 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
|
D | cuda_dnn.cc | 2985 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution() argument 2994 input_descriptor, in DoPrepareForConvolution() 2995 ToCudnnDataType(element_type, input_descriptor.layout())); in DoPrepareForConvolution() 3041 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve() argument 3049 CudnnTensorDescriptor input_nd(input_descriptor, cudnn_type); in DoConvolve() 4005 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBiasImpl() argument 4010 CudnnTensorDescriptor input_nd(input_descriptor, cudnn_type); in DoConvolveBackwardBiasImpl() 4026 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias() argument 4031 DoConvolveBackwardBiasImpl(stream, input_descriptor, input_data, in DoConvolveBackwardBias() 4037 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias() argument [all …]
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 1296 Stream* stream, const BatchDescriptor& input_descriptor, 1305 bool DoConvolve(Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolve() argument 1317 ToDataType<OutputType>::value, stream, input_descriptor, in DoConvolve() 1332 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, 1350 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 1361 Stream* stream, const dnn::BatchDescriptor& input_descriptor, 1379 Stream* stream, const BatchDescriptor& input_descriptor, 1413 const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardData() argument 1421 ToDataType<ElementType>::value, stream, input_descriptor, in DoConvolveBackwardData() 1455 Stream* stream, const BatchDescriptor& input_descriptor, in DoConvolveBackwardFilter() argument [all …]
|
D | stream.h | 297 Stream &ThenConvolve(const dnn::BatchDescriptor &input_descriptor, 306 const dnn::BatchDescriptor &input_descriptor, 316 const dnn::BatchDescriptor &input_descriptor, 327 const dnn::BatchDescriptor &input_descriptor, in ConvolveWithAlgorithm() argument 340 dnn::ConvolutionKind::FORWARD, this, input_descriptor, input_data, in ConvolveWithAlgorithm() 346 dnn::ToDataType<OutputType>::value, this, input_descriptor, in ConvolveWithAlgorithm() 426 const dnn::BatchDescriptor &input_descriptor, 442 const dnn::BatchDescriptor &input_descriptor, in ConvolveBackwardDataWithAlgorithm() argument 451 dnn::ConvolutionKind::BACKWARD_DATA, this, input_descriptor, in ConvolveBackwardDataWithAlgorithm() 459 dnn::ToDataType<ElementType>::value, this, input_descriptor, in ConvolveBackwardDataWithAlgorithm() [all …]
|
D | stream.cc | 603 const dnn::BatchDescriptor &input_descriptor, in ThenConvolve() argument 612 input_descriptor, input_data, filter_descriptor, filter_data, in ThenConvolve() 622 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveQuantized() argument 630 VLOG_CALL(PARAM(input_descriptor), PARAM(input_data), in ThenConvolveQuantized() 637 this, input_descriptor, input_data, filter_descriptor, in ThenConvolveQuantized() 649 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveQuantized() argument 657 VLOG_CALL(PARAM(input_descriptor), PARAM(input_data), in ThenConvolveQuantized() 664 this, input_descriptor, input_data, filter_descriptor, in ThenConvolveQuantized() 702 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardBiasImpl() argument 706 VLOG_CALL(PARAM(input_descriptor), PARAM(input_data), PARAM(bias_descriptor), in ThenConvolveBackwardBiasImpl() [all …]
|
D | stream_executor_pimpl.cc | 279 const dnn::BatchDescriptor &input_descriptor, DeviceMemoryBase input_data, in GetMIOpenConvolveAlgorithms() argument 291 kind, element_type, stream, input_descriptor, input_data, in GetMIOpenConvolveAlgorithms()
|
D | stream_executor_pimpl.h | 372 const dnn::BatchDescriptor &input_descriptor, DeviceMemoryBase input_data,
|