| /external/tensorflow/tensorflow/core/kernels/ |
| fused_batch_norm_op.cc | 303 se::dnn::BatchDescriptor x_desc; local
304 x_desc.set_count(batch_size)
362 estimated_variance_ptr, x_desc, scale_offset_desc,
457 se::dnn::BatchDescriptor x_desc; local
458 x_desc.set_count(batch_size)
488 x_desc, scale_offset_desc, static_cast<double>(epsilon),
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 222 const dnn::BatchDescriptor& x_desc,
235 const dnn::BatchDescriptor& x_desc,
247 const dnn::BatchDescriptor& x_desc,
256 const dnn::BatchDescriptor& x_desc,
563 const dnn::BatchDescriptor& x_desc,
576 const DeviceMemory<U>& inv_var, const dnn::BatchDescriptor& x_desc,
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.h | 213 const dnn::BatchDescriptor& x_desc,
226 const dnn::BatchDescriptor& x_desc,
238 const dnn::BatchDescriptor& x_desc,
247 const dnn::BatchDescriptor& x_desc,
[all...] |
| rocm_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | [all...] |
| stream.cc | 344 const dnn::BatchDescriptor &x_desc,
351 VLOG_CALL(PARAM(x), PARAM(scale), PARAM(offset), PARAM(x_desc),
356 this, x, scale, offset, estimated_mean, estimated_variance, x_desc,
370 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc,
374 VLOG_CALL(PARAM(y_backprop), PARAM(x), PARAM(scale), PARAM(x_desc),
380 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc,
394 const dnn::BatchDescriptor &x_desc,
401 VLOG_CALL(PARAM(x), PARAM(scale), PARAM(offset), PARAM(x_desc),
406 this, x, scale, offset, estimated_mean, estimated_variance, x_desc,
421 const dnn::BatchDescriptor &x_desc,
[all...] |
| stream.h | 237 const dnn::BatchDescriptor &x_desc,
248 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc,
258 const dnn::BatchDescriptor &x_desc,
270 const dnn::BatchDescriptor &x_desc,
[all...] |
