| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_batchnorm_thunk.cc | 35 dnn::BatchDescriptor /*scale_offset_desc*/>
69 dnn::BatchDescriptor scale_offset_desc; local
70 scale_offset_desc.set_layout(dnn::DataLayout::kBatchDepthYX)
76 return std::make_pair(input_desc, scale_offset_desc);
105 dnn::BatchDescriptor scale_offset_desc; local
106 std::tie(operand_desc, scale_offset_desc) =
117 scale_offset_desc, // local
164 dnn::BatchDescriptor scale_offset_desc; local
168 std::tie(operand_desc, scale_offset_desc) = MakeDescriptors(
186 scale_offset_desc, // local
246 dnn::BatchDescriptor scale_offset_desc; local
268 operand_desc, scale_offset_desc, epsilon_, &output_grad_data, local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| fused_batch_norm_op.cc | 289 perftools::gputools::dnn::BatchDescriptor scale_offset_desc; local
290 scale_offset_desc.set_count(1)
341 estimated_variance_ptr, x_desc, scale_offset_desc,
443 perftools::gputools::dnn::BatchDescriptor scale_offset_desc; local
444 scale_offset_desc.set_count(1)
467 x_desc, scale_offset_desc, static_cast<double>(epsilon),
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 206 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
219 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
231 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
240 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | 903 // scale_offset_desc: dimensions of scale and offset.
[all...] |
| stream.cc | 320 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
327 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(y));
332 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean,
346 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
350 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(x_backprop),
355 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc,
370 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
377 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(y));
382 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean,
397 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon
[all...] |
| stream.h | 227 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
238 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
248 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
260 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
[all...] |
