Searched defs:y_backprop (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 164 typename TTypes<T, 4>::ConstTensor y_backprop( in operator ()() local
615 const Tensor& y_backprop = context->input(0); in Compute() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 2453 ShapeHandle y_backprop; in __anon3e672dd83f02() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3280 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward()
3295 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward()
3312 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 825 ShapeHandle y_backprop; in FusedBatchNormGradShape() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3085 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward()
3099 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward()
3115 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 199 Output y_backprop = in SimpleFusedBatchNormGrad() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 368 const DeviceMemory<float> &y_backprop, const DeviceMemory<float> &x, in ThenBatchNormalizationBackward()
418 const DeviceMemory<Eigen::half> &y_backprop, in ThenBatchNormalizationBackward()
|