Home
last modified time | relevance | path

Searched refs:scale_offset_desc (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc68 dnn::BatchDescriptor scale_offset_desc; in MakeDescriptors() local
69 scale_offset_desc.set_layout(dnn::DataLayout::kBatchDepthYX) in MakeDescriptors()
75 return std::make_pair(input_desc, scale_offset_desc); in MakeDescriptors()
105 dnn::BatchDescriptor scale_offset_desc; in ExecuteOnStream() local
106 std::tie(operand_desc, scale_offset_desc) = in ExecuteOnStream()
118 scale_offset_desc, // in ExecuteOnStream()
167 dnn::BatchDescriptor scale_offset_desc; in ExecuteOnStream() local
171 std::tie(operand_desc, scale_offset_desc) = MakeDescriptors( in ExecuteOnStream()
190 scale_offset_desc, // in ExecuteOnStream()
251 dnn::BatchDescriptor scale_offset_desc; in ExecuteOnStream() local
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h223 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
236 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
248 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
257 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
564 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
577 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
Dcuda_dnn.cc3184 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward() argument
3193 offset, estimated_mean, estimated_variance, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3205 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward() argument
3214 estimated_mean, estimated_variance, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3228 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForwardImpl() argument
3235 scale_offset_desc, ToCudnnDataType(scale_data_type)); in DoBatchNormalizationForwardImpl()
3284 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward() argument
3289 x, scale, mean, inv_var, x_desc, scale_offset_desc, in DoBatchNormalizationBackward()
3299 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward() argument
3304 x, scale, mean, inv_var, x_desc, scale_offset_desc, in DoBatchNormalizationBackward()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc310 se::dnn::BatchDescriptor scale_offset_desc; in operator ()() local
311 scale_offset_desc.set_count(1) in operator ()()
362 estimated_variance_ptr, x_desc, scale_offset_desc, in operator ()()
464 se::dnn::BatchDescriptor scale_offset_desc; in operator ()() local
465 scale_offset_desc.set_count(1) in operator ()()
488 x_desc, scale_offset_desc, static_cast<double>(epsilon), in operator ()()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h214 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
227 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
239 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
248 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
636 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
649 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
Drocm_dnn.cc3002 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward() argument
3010 estimated_mean, estimated_variance, x_desc, scale_offset_desc, epsilon, y, in DoBatchNormalizationForward()
3021 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward() argument
3029 estimated_mean, estimated_variance, x_desc, scale_offset_desc, epsilon, y, in DoBatchNormalizationForward()
3042 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForwardImpl() argument
3052 scale_offset_desc, ToMIOpenDataType(scale_data_type)}; in DoBatchNormalizationForwardImpl()
3089 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward() argument
3094 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward()
3103 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward() argument
3108 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h996 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
1013 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
1044 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
1058 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
Dstream.cc345 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward() argument
352 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(y)); in ThenBatchNormalizationForward()
357 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean, in ThenBatchNormalizationForward()
371 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward() argument
375 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(x_backprop), in ThenBatchNormalizationBackward()
380 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward()
395 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward() argument
402 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(y)); in ThenBatchNormalizationForward()
407 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean, in ThenBatchNormalizationForward()
422 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward() argument
[all …]
Dstream.h238 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
249 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
259 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
271 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,