| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_FusedBatchNormGrad.pbtxt | 2 graph_op_name: "FusedBatchNormGrad"
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 4 name: "nn.FusedBatchNormGrad"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_FusedBatchNormGrad.pbtxt | 2 graph_op_name: "FusedBatchNormGrad"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op_test.cc | 240 static Graph* FusedBatchNormGrad(int n, int h, int w, int c, bool is_training, in FusedBatchNormGrad() function
329 static void BM_NAME(FusedBatchNormGrad, N, H, W, C, T, IS_TRAINING, FORMAT, \
333 FusedBatchNormGrad<T>(N, H, W, C, IS_TRAINING, FORMAT_##FORMAT), \
339 BM_NAME(FusedBatchNormGrad, N, H, W, C, T, IS_TRAINING, FORMAT, DEVICE)) \
|
| D | fused_batch_norm_op.cc | 89 struct FusedBatchNormGrad;
341 struct FusedBatchNormGrad<CPUDevice, T, U> { struct
984 struct FusedBatchNormGrad<GPUDevice, T, U> { struct
1476 functor::FusedBatchNormGrad<Device, T, U>()( in ComputeWithReservedSpace()
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | keras_imagenet_main_graph_mode.golden_summary | 30 FusedBatchNormGrad 53
|
| D | keras_imagenet_main_graph_mode.pbtxt | 44577 name: "training/SGD/gradients/bn5c_branch2c_1/FusedBatchNorm_grad/FusedBatchNormGrad"
44578 op: "FusedBatchNormGrad"
44630 input: "training/SGD/gradients/bn5c_branch2c_1/FusedBatchNorm_grad/FusedBatchNormGrad:2"
44658 input: "training/SGD/gradients/bn5c_branch2c_1/FusedBatchNorm_grad/FusedBatchNormGrad:1"
44685 input: "training/SGD/gradients/bn5c_branch2c_1/FusedBatchNorm_grad/FusedBatchNormGrad"
44746 input: "training/SGD/gradients/bn5c_branch2c_1/FusedBatchNorm_grad/FusedBatchNormGrad"
44874 name: "training/SGD/gradients/bn5c_branch2b_1/FusedBatchNorm_grad/FusedBatchNormGrad"
44875 op: "FusedBatchNormGrad"
44953 input: "training/SGD/gradients/bn5c_branch2b_1/FusedBatchNorm_grad/FusedBatchNormGrad:2"
44981 input: "training/SGD/gradients/bn5c_branch2b_1/FusedBatchNorm_grad/FusedBatchNormGrad:1"
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | FusedBatchNormGrad.pbtxt | 2 name: "FusedBatchNormGrad"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | FusedBatchNormGrad.pbtxt | 2 name: "FusedBatchNormGrad"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNormGrad.pbtxt | 2 graph_op_name: "FusedBatchNormGrad"
|
| /external/tensorflow/tensorflow/core/profiler/g3doc/ |
| D | profile_memory.md | 27 FusedBatchNormGrad 10802.14MB (72.98%, 8.98%)
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 192 * `raw_ops.FusedBatchNormGrad`
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 218 ops::FusedBatchNormGrad::Attrs attrs; in SimpleFusedBatchNormGrad()
221 ops::FusedBatchNormGrad(s->WithOpName("FusedBatchNormGrad"), y_backprop, in SimpleFusedBatchNormGrad()
|
| D | auto_mixed_precision_test.cc | 520 Output bng1 = ops::FusedBatchNormGrad( in TEST_F()
522 fbn1_rs2, ops::FusedBatchNormGrad::DataFormat("NHWC")) in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | cpu_supported_ops.md | 90 `FusedBatchNormGrad` | `T={float}`
|
| D | gpu_supported_ops.md | 90 `FusedBatchNormGrad` | `T={float}`
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1781 ### tf.FusedBatchNormGrad
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 193 …%0:5 = "tf.FusedBatchNormGrad"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_fo…
208 …%0:5 = "tf.FusedBatchNormGrad"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_fo…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1720 name: "FusedBatchNormGrad"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1720 name: "FusedBatchNormGrad"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 12836 name: "FusedBatchNormGrad"
12915 name: "FusedBatchNormGrad"
|
| D | ops_history.v2.pbtxt | 27225 name: "FusedBatchNormGrad"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 27092 func FusedBatchNormGrad(scope *Scope, y_backprop tf.Output, x tf.Output, scale tf.Output, reserve_s… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 17470 name: "FusedBatchNormGrad"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 4572 def TF_FusedBatchNormGradOp : TF_Op<"FusedBatchNormGrad", [NoSideEffect]> {
|