| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 286 typename TTypes<U>::ConstVec estimated_mean(estimated_mean_input.vec<U>()); in operator ()() local
314 estimated_mean.reshape(one_by_depth).broadcast(bcast_spec); in operator ()()
325 batch_mean.device(d) = estimated_mean; in operator ()()
761 const Tensor& estimated_mean, in operator ()()
840 estimated_mean.vec<U>(), estimated_variance.vec<U>(), in operator ()()
847 estimated_mean.vec<U>(), estimated_variance.vec<U>(), in operator ()()
908 StreamExecutorUtil::AsDeviceMemory<U>(estimated_mean); in operator ()()
932 if (!batch_mean->SharesBufferWith(estimated_mean) && in operator ()()
938 estimated_mean.NumElements() * sizeof(U)) in operator ()()
1164 typename TTypes<U>::ConstVec estimated_mean, \
[all …]
|
| D | fused_eigen_output_kernels.h | 398 const Tensor& estimated_mean = context->input(4);
407 if (estimated_mean.dims() != 1)
409 estimated_mean.shape().DebugString());
420 args->estimated_mean_data = data_ptr(estimated_mean);
|
| D | fused_batch_norm_op.h | 54 typename TTypes<U>::ConstVec estimated_mean,
|
| D | fused_batch_norm_op.cu.cc | 323 typename TTypes<U>::ConstVec estimated_mean, in operator ()()
355 estimated_mean.data(), estimated_variance.data(), side_input.data(), \ in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 221 const DeviceMemory<float>& estimated_mean,
235 const DeviceMemory<float>& estimated_mean,
592 const DeviceMemory<U>& estimated_mean,
|
| D | cuda_dnn.cc | 3450 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward() argument
3463 offset, estimated_mean, estimated_variance, side_input, x_desc, in DoBatchNormalizationForward()
3473 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward() argument
3486 estimated_mean, estimated_variance, side_input, x_desc, in DoBatchNormalizationForward()
3498 const DeviceMemory<U>& estimated_mean, in DoBatchNormalizationForwardImpl() argument
3636 scale.opaque(), offset.opaque(), estimated_mean.opaque(), maybe_inv_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 259 const DeviceMemory<float>& estimated_mean,
273 const DeviceMemory<float>& estimated_mean,
708 const DeviceMemory<U>& estimated_mean,
|
| D | rocm_dnn.cc | 3524 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward() argument
3536 estimated_mean, estimated_variance, side_input, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3544 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward() argument
3556 estimated_mean, estimated_variance, side_input, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3566 const DeviceMemory<U>& estimated_mean, in DoBatchNormalizationForwardImpl() argument
3599 const_cast<void*>(estimated_mean.opaque()), in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1014 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward() argument
1033 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward() argument
|
| D | stream.h | 249 const DeviceMemory<float> &estimated_mean,
274 const DeviceMemory<float> &estimated_mean,
|
| D | stream.cc | 345 const DeviceMemory<float> &estimated_mean, in ThenBatchNormalizationForward() argument
360 this, x, scale, offset, estimated_mean, estimated_variance, side_input, in ThenBatchNormalizationForward()
396 const DeviceMemory<float> &estimated_mean, in ThenBatchNormalizationForward() argument
411 this, x, scale, offset, estimated_mean, estimated_variance, side_input, in ThenBatchNormalizationForward()
|