Home
last modified time | relevance | path

Searched refs:input_descriptor (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc181 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/
Drocm_dnn.cc320 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 …]
Drocm_dnn.h255 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/
Dcuda_dnn.h264 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,
Dcuda_dnn.cc2627 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/
Ddnn.h1242 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 …]
Dstream.cc594 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 …]
Dstream.h276 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 …]