Home
last modified time | relevance | path

Searched defs:scale_backprop (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op_test.cc162 Tensor scale_backprop; in VerifyTensorsCloseForGrad() local
385 auto scale_backprop = ops::Identity(root.WithOpName("scale_backprop"), in VerifyFusedBatchNormGradWithConv2D() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc272 xla::XlaOp scale_backprop; in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cu.cc49 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>()); in operator ()() local
Dfused_batch_norm_op.cc530 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>()); in operator ()() local
1451 Tensor* scale_backprop = nullptr; in ComputeWithReservedSpace() local
Dfused_batch_norm_ex_op_test.cc62 Tensor scale_backprop; member
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc276 auto scale_backprop = ops::Identity(scope.WithOpName("scale_backprop"), in CreateSimpleFusedBatchNormGrad() local
1207 auto* scale_backprop = context.graph_view->GetNode("scale_backprop"); in TEST_F() local
1270 auto* scale_backprop = context.graph_view->GetNode("scale_backprop"); in TEST_F() local
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1070 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
1087 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
Dstream.cc375 DeviceMemory<float> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward()
427 DeviceMemory<Eigen::half> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3648 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3666 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3685 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3616 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3632 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3649 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1834 Value x_backprop, scale_backprop, offset_backprop; in matchAndRewrite() local