Searched defs:offset_backprop (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 251 xla::XlaOp offset_backprop; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cu.cc | 46 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
|
| D | fused_batch_norm_op.cc | 301 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
1330 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
1082 auto* offset_backprop = context.graph_view->GetNode("offset_backprop"); in TEST_F() local
1145 auto* offset_backprop = context.graph_view->GetNode("offset_backprop"); in TEST_F() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3376 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3392 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3409 DeviceMemory<U>* offset_backprop) { in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3669 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3687 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward()
3706 DeviceMemory<U>* offset_backprop, DeviceMemory<uint8>* reserve_space_data, 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 | 380 DeviceMemory<float> *offset_backprop, in ThenBatchNormalizationBackward()
440 DeviceMemory<float> *offset_backprop, in ThenBatchNormalizationBackward()
|