Home
last modified time | relevance | path

Searched refs:BatchDescriptor (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h50 bool IsSame(const dnn::BatchDescriptor& input_dimensions,
51 const dnn::BatchDescriptor& output_dimensions,
61 bool find(const void* p, const dnn::BatchDescriptor& input_dimensions,
62 const dnn::BatchDescriptor& output_dimensions,
65 void insert(const void* p, const dnn::BatchDescriptor& input_dimensions,
66 const dnn::BatchDescriptor& output_dimensions,
243 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
246 const dnn::BatchDescriptor& output_descriptor,
256 dnn::DataType output_type, const dnn::BatchDescriptor& input_descriptor,
258 const dnn::BatchDescriptor& output_descriptor,
[all …]
Drocm_dnn.cc68 using dnn::BatchDescriptor;
651 ScopedTensorDescriptor(const BatchDescriptor& batch_descriptor, in ScopedTensorDescriptor()
2905 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution()
2907 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
2956 BatchDescriptor input_descriptor, in RocmConvRunner()
2957 BatchDescriptor output_descriptor, in RocmConvRunner()
3114 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve()
3116 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
3149 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in GetConvolveRunners()
3151 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetConvolveRunners()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h42 port::ArraySlice<dnn::BatchDescriptor>; // non-absl ok
223 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
226 const dnn::BatchDescriptor& output_descriptor,
236 dnn::DataType output_type, const dnn::BatchDescriptor& input_descriptor,
238 const dnn::BatchDescriptor& output_descriptor,
246 const dnn::BatchDescriptor& input_descriptor,
248 const dnn::BatchDescriptor& bias_descriptor,
249 const dnn::BatchDescriptor& output_descriptor,
261 const dnn::BatchDescriptor& input_descriptor,
263 const dnn::BatchDescriptor& bias_descriptor,
[all …]
Dcuda_dnn.cc614 CudnnTensorDescriptor(const dnn::BatchDescriptor& batch_descriptor, in CudnnTensorDescriptor()
3455 const dnn::BatchDescriptor& tensor, dnn::DataType element_type) { in GetTensorVectorSizeAndDim()
3506 const dnn::BatchDescriptor& input_descriptor, in GetCudnnOperationGraph()
3508 const dnn::BatchDescriptor& output_descriptor, in GetCudnnOperationGraph()
3632 const dnn::BatchDescriptor& input_descriptor, in GetCudnnFusedOperationGraph()
3634 dnn::BatchDescriptor bias_descriptor, in GetCudnnFusedOperationGraph()
3635 const dnn::BatchDescriptor& output_descriptor, in GetCudnnFusedOperationGraph()
3958 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution()
3960 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
4259 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.h238 class BatchDescriptor {
242 BatchDescriptor();
243 explicit BatchDescriptor(int ndims);
246 void CloneFrom(const BatchDescriptor& other);
291 BatchDescriptor& set_count(int64_t value) { in set_count()
295 BatchDescriptor& set_feature_map_count(int64_t value) { in set_feature_map_count()
299 BatchDescriptor& set_height(int64_t value) { in set_height()
303 BatchDescriptor& set_width(int64_t value) { in set_width()
307 BatchDescriptor& set_spatial_dim(DimIndex dim, int64_t value) { in set_spatial_dim()
311 BatchDescriptor& set_value_max(float value) { in set_value_max()
[all …]
Ddnn.cc119 Stream* /*stream*/, const dnn::BatchDescriptor& /*input_descriptor*/, in GetConvolveRunners() argument
123 const dnn::BatchDescriptor& /*output_descriptor*/, in GetConvolveRunners()
135 dnn::DataType output_type, const dnn::BatchDescriptor& input_descriptor, in ConvolveRunnerFromDesc()
137 const dnn::BatchDescriptor& output_descriptor, in ConvolveRunnerFromDesc()
147 const dnn::BatchDescriptor& input_descriptor, in GetFusedConvolveRunners()
149 const dnn::BatchDescriptor& bias_descriptor, in GetFusedConvolveRunners()
150 const dnn::BatchDescriptor& output_descriptor, in GetFusedConvolveRunners()
163 const dnn::BatchDescriptor& input_descriptor, in FusedConvolveRunnerFromDesc()
165 const dnn::BatchDescriptor& bias_descriptor, in FusedConvolveRunnerFromDesc()
166 const dnn::BatchDescriptor& output_descriptor, in FusedConvolveRunnerFromDesc()
[all …]
Dstream.h67 class BatchDescriptor; variable
257 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc,
258 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
270 const DeviceMemory<float> &y, const dnn::BatchDescriptor &x_desc,
271 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
284 const dnn::BatchDescriptor &x_desc,
285 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
298 const dnn::BatchDescriptor &x_desc,
299 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
307 Stream &ThenConvolve(const dnn::BatchDescriptor &input_descriptor,
[all …]
Dlazy_op_runner.h123 const BatchDescriptor& input_descriptor;
125 const BatchDescriptor& output_descriptor;
147 const BatchDescriptor& input_descriptor;
149 const BatchDescriptor& bias_descriptor;
150 const BatchDescriptor& output_descriptor;
Dstream.cc39 std::string ToVlogString(const dnn::BatchDescriptor &descriptor) { in ToVlogString()
333 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
334 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward()
359 const DeviceMemory<float> &y, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward()
360 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward()
387 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
388 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward()
414 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward()
415 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward()
438 const dnn::BatchDescriptor &input_descriptor, in ThenConvolve()
[all …]
Dstream_executor_pimpl.h375 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
378 const dnn::BatchDescriptor& output_descriptor,
389 const dnn::BatchDescriptor& input_descriptor,
391 const dnn::BatchDescriptor& bias_descriptor,
392 const dnn::BatchDescriptor& output_descriptor,
401 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
404 const dnn::BatchDescriptor& output_descriptor,
Dstream_executor_pimpl.cc296 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in GetConvolveRunners()
298 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetConvolveRunners()
319 const dnn::BatchDescriptor& input_descriptor, in GetFusedConvolveRunners()
321 const dnn::BatchDescriptor& bias_descriptor, in GetFusedConvolveRunners()
322 const dnn::BatchDescriptor& output_descriptor, in GetFusedConvolveRunners()
339 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in GetMIOpenConvolveAlgorithms()
341 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetMIOpenConvolveAlgorithms()
Ddnn.proto69 // BatchDescriptor::value_max().
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.cc110 const se::dnn::BatchDescriptor& input_desc, in AutotuneFusedConv()
112 const se::dnn::BatchDescriptor& bias_desc, in AutotuneFusedConv()
113 const se::dnn::BatchDescriptor& output_desc, in AutotuneFusedConv()
225 const se::dnn::BatchDescriptor& input_desc,
227 const se::dnn::BatchDescriptor& bias_desc,
228 const se::dnn::BatchDescriptor& output_desc,
241 const se::dnn::BatchDescriptor& input_desc,
243 const se::dnn::BatchDescriptor& bias_desc,
244 const se::dnn::BatchDescriptor& output_desc,
258 const se::dnn::BatchDescriptor& input_desc,
[all …]
Dconv_ops_gpu.h105 const se::dnn::BatchDescriptor& input_desc,
107 const se::dnn::BatchDescriptor& bias_desc,
108 const se::dnn::BatchDescriptor& output_desc,
121 se::dnn::ConvolutionKind kind, const se::dnn::BatchDescriptor& input_desc,
125 const se::dnn::BatchDescriptor& output_desc, se::DeviceMemory<T> output_ptr,
164 const se::dnn::BatchDescriptor& input_desc, in LaunchAutotunedConv()
169 const se::dnn::BatchDescriptor& output_desc, in LaunchAutotunedConv()
Dcudnn_pooling_gpu.cc74 se::dnn::BatchDescriptor input_desc(3); in Compute()
78 se::dnn::BatchDescriptor output_desc(3); in Compute()
202 se::dnn::BatchDescriptor orig_output_desc(3); in Compute()
207 se::dnn::BatchDescriptor orig_input_desc(3); in Compute()
Dgpu_utils.cc142 const se::dnn::BatchDescriptor& input_desc, in LogConvAutotuneResults()
144 const se::dnn::BatchDescriptor& output_desc, in LogConvAutotuneResults()
185 const se::dnn::BatchDescriptor& input_desc, in LogFusedConvForwardAutotuneResults()
187 const se::dnn::BatchDescriptor& output_desc, in LogFusedConvForwardAutotuneResults()
Dgpu_utils.h261 const se::dnn::BatchDescriptor& input_desc,
263 const se::dnn::BatchDescriptor& output_desc,
273 const se::dnn::BatchDescriptor& input_desc,
275 const se::dnn::BatchDescriptor& output_desc,
Dpooling_ops_common.cc365 se::dnn::BatchDescriptor input_desc; in Compute()
372 se::dnn::BatchDescriptor output_desc; in Compute()
686 se::dnn::BatchDescriptor orig_output_desc; in Compute()
693 se::dnn::BatchDescriptor orig_input_desc; in Compute()
Dlrn_op.cc203 se::dnn::BatchDescriptor dimensions_desc; in launch()
256 perftools::gputools::dnn::BatchDescriptor dimensions_desc; in launch()
495 se::dnn::BatchDescriptor dimensions_desc; in launch()
570 perftools::gputools::dnn::BatchDescriptor dimensions_desc; in launch()
Dconv_ops_fused_impl.h492 se::dnn::BatchDescriptor input_desc;
504 se::dnn::BatchDescriptor bias_desc;
518 se::dnn::BatchDescriptor output_desc;
Dconv_grad_input_ops.cc245 se::dnn::BatchDescriptor input_desc; in operator ()()
251 se::dnn::BatchDescriptor output_desc; in operator ()()
Dconv_ops_3d.cc405 se::dnn::BatchDescriptor input_desc(3); in launch()
412 se::dnn::BatchDescriptor output_desc(3); in launch()
Dfused_batch_norm_op.cc900 se::dnn::BatchDescriptor x_desc; in operator ()()
907 se::dnn::BatchDescriptor scale_offset_desc; in operator ()()
1095 se::dnn::BatchDescriptor x_desc; in operator ()()
1102 se::dnn::BatchDescriptor scale_offset_desc; in operator ()()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc30 se::dnn::BatchDescriptor GetBiasDescriptor(const GpuConvConfig& config) { in GetBiasDescriptor()
31 se::dnn::BatchDescriptor result(config.output_descriptor.ndims()); in GetBiasDescriptor()
61 using se::dnn::BatchDescriptor;
122 BatchDescriptor bias_desc = GetBiasDescriptor(*params.config); in RunGpuConvForwardActivation()
393 BatchDescriptor& input_descriptor = config.input_descriptor; in GetGpuConvConfig()
394 input_descriptor = BatchDescriptor(effective_num_dimensions); in GetGpuConvConfig()
437 BatchDescriptor& output_descriptor = config.output_descriptor; in GetGpuConvConfig()
438 output_descriptor = BatchDescriptor(effective_num_dimensions); in GetGpuConvConfig()
Dgpu_conv_runner.h61 se::dnn::BatchDescriptor input_descriptor;
63 se::dnn::BatchDescriptor output_descriptor;
215 se::dnn::BatchDescriptor GetBiasDescriptor(const GpuConvConfig& config);

12