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 | 164 Tensor offset_backprop; in VerifyTensorsCloseForGrad() local
390 auto offset_backprop = ops::Identity( in VerifyFusedBatchNormGradWithConv2D() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 274 xla::XlaOp offset_backprop; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cu.cc | 51 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
|
| D | fused_batch_norm_op.cc | 414 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
1501 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 | 1180 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
1196 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
|
| D | stream.cc | 375 DeviceMemory<float> *offset_backprop, in ThenBatchNormalizationBackward()
427 DeviceMemory<float> *offset_backprop, in ThenBatchNormalizationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3612 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3628 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3645 DeviceMemory<U>* offset_backprop) { in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 4760 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
4778 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
4797 DeviceMemory<U>* offset_backprop, DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2444 Value x_backprop, scale_backprop, offset_backprop; in matchAndRewrite() local
|