Searched defs:offset_backprop (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 163 Tensor offset_backprop; in VerifyTensorsCloseForGrad() local
387 auto offset_backprop = ops::Identity( in VerifyFusedBatchNormGradWithConv2D() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 273 xla::XlaOp offset_backprop; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cu.cc | 50 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
|
| D | fused_batch_norm_op.cc | 393 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
1454 Tensor* offset_backprop = nullptr; in ComputeWithReservedSpace() local
|
| D | fused_batch_norm_ex_op_test.cc | 63 Tensor offset_backprop; member
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 278 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
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1071 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
1087 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
|
| D | stream.cc | 376 DeviceMemory<float> *offset_backprop, in ThenBatchNormalizationBackward()
428 DeviceMemory<float> *offset_backprop, in ThenBatchNormalizationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3649 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3667 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3686 DeviceMemory<U>* offset_backprop, DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3617 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3633 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3650 DeviceMemory<U>* offset_backprop) { in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1834 Value x_backprop, scale_backprop, offset_backprop; in matchAndRewrite() local
|