Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc89 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()
Dgpu_conv_runner.h70 se::dnn::BatchDescriptor input_descriptor; member
Dgpu_conv_algorithm_picker.cc143 kind, dtype, stream, params.config.input_descriptor, params.input_buf, in GetMIOpenAlgorithms()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc435 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 …]
Drocm_dnn.h236 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/
Dcuda_dnn.h271 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,
Dcuda_dnn.cc2985 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/
Ddnn.h1296 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 …]
Dstream.h297 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 …]
Dstream.cc603 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 …]
Dstream_executor_pimpl.cc279 const dnn::BatchDescriptor &input_descriptor, DeviceMemoryBase input_data, in GetMIOpenConvolveAlgorithms() argument
291 kind, element_type, stream, input_descriptor, input_data, in GetMIOpenConvolveAlgorithms()
Dstream_executor_pimpl.h372 const dnn::BatchDescriptor &input_descriptor, DeviceMemoryBase input_data,