Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_batch_norm_op_test.cc164 Tensor offset_backprop; in VerifyTensorsCloseForGrad() local
390 auto offset_backprop = ops::Identity( in VerifyFusedBatchNormGradWithConv2D() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbatch_norm_op.cc277 xla::XlaOp offset_backprop; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_op.cc414 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
1583 Tensor* offset_backprop = nullptr; in ComputeWithReservedSpace() local
H A Dfused_batch_norm_op.cu.cc51 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
H A Dfused_batch_norm_ex_op_test.cc63 Tensor offset_backprop; member
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer_test.cc278 auto offset_backprop = ops::Identity(scope.WithOpName("offset_backprop"), in CreateSimpleFusedBatchNormGrad() local
1212 auto* offset_backprop = context.graph_view->GetNode("offset_backprop"); in TEST_F() local
1275 auto* offset_backprop = context.graph_view->GetNode("offset_backprop"); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h1224 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
1243 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
H A Dstream.cc362 DeviceMemory<float> *scale_backprop, DeviceMemory<float> *offset_backprop, in ThenBatchNormalizationBackward()
417 DeviceMemory<float> *scale_backprop, DeviceMemory<float> *offset_backprop, in ThenBatchNormalizationBackward()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc3734 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3752 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3770 DeviceMemory<U>* offset_backprop) { in DoBatchNormalizationBackwardImpl()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc5612 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
5633 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
5655 DeviceMemory<U>* scale_backprop, DeviceMemory<U>* offset_backprop, in DoBatchNormalizationBackwardImpl()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc2239 Value x_backprop, scale_backprop, offset_backprop; in matchAndRewrite() local