Searched defs:scale_backprop (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 250 xla::XlaOp scale_backprop; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cu.cc | 45 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>()); in operator ()() local
|
| D | fused_batch_norm_op.cc | 438 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>()); in operator ()() local
1327 Tensor* scale_backprop = nullptr; in ComputeWithReservedSpace() local
|
| D | fused_batch_norm_ex_op_test.cc | 62 Tensor scale_backprop; member
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 276 auto scale_backprop = ops::Identity(scope.WithOpName("scale_backprop"), in CreateSimpleFusedBatchNormGrad() local
1077 auto* scale_backprop = context.graph_view->GetNode("scale_backprop"); in TEST_F() local
1140 auto* scale_backprop = context.graph_view->GetNode("scale_backprop"); in TEST_F() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3375 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3391 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3408 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3668 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3686 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3705 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 938 Value x_backprop, scale_backprop, offset_backprop; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 379 DeviceMemory<float> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward()
439 DeviceMemory<Eigen::half> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward()
|