| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | FusedBatchNorm.pbtxt | 32 name: "batch_variance"
111 name: "batch_variance"
|
| D | FusedBatchNormV2.pbtxt | 32 name: "batch_variance"
122 name: "batch_variance"
|
| D | FusedBatchNormV3.pbtxt | 32 name: "batch_variance"
133 name: "batch_variance"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | FusedBatchNorm.pbtxt | 32 name: "batch_variance"
111 name: "batch_variance"
|
| D | FusedBatchNormV2.pbtxt | 32 name: "batch_variance"
122 name: "batch_variance"
|
| D | FusedBatchNormV3.pbtxt | 32 name: "batch_variance"
126 name: "batch_variance"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 53 Tensor batch_variance; member
213 forward->batch_variance = out_tensors[2]; in RunFusedBatchNorm()
325 forward->batch_variance = out_tensors[2]; in RunFusedBatchNormEx()
399 {fbn_forward.batch_variance, fbn_ex_forward.batch_variance}, in VerifyTensorsNear()
|
| D | fused_batch_norm_op.cc | 183 Eigen::Tensor<U, 1, Eigen::RowMajor> batch_variance(depth); in operator ()() local
189 batch_variance.device(d) = in operator ()()
191 auto scaling_factor = ((batch_variance + epsilon).rsqrt() * scale) in operator ()()
202 saved_batch_var.device(d) = batch_variance; in operator ()()
204 new_variance.device(d) = batch_variance * rest_size_adjust; in operator ()()
208 saved_batch_var.device(d) = batch_variance; in operator ()()
212 (exponential_avg_factor * rest_size_adjust) * batch_variance; in operator ()()
291 typename TTypes<U>::Vec batch_variance(batch_var_output->vec<U>()); in operator ()() local
326 batch_variance.device(d) = estimated_variance; in operator ()()
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | normalization.py | 184 batch_mean, batch_variance = nn_impl.moments_v2(
201 (batch_variance +
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNorm.pbtxt | 49 name: "batch_variance"
|
| D | api_def_FusedBatchNormV3.pbtxt | 49 name: "batch_variance"
|
| D | api_def_FusedBatchNormV2.pbtxt | 49 name: "batch_variance"
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | check_numerics_callback_test.py | 426 _, batch_mean, batch_variance, _, _ = self.evaluate(batch_norm_res)
429 self.assertTrue(np.isnan(batch_variance.squeeze()))
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | layout_optimization_to_nhwc.mlir | 60 %6, %batch_mean, %batch_variance, %reserved_1, %reserved_2, %reserved_3 =
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 254 ops::Identity(root.WithOpName("y_batch_var"), bn.batch_variance); in VerifyFusedBatchNorm()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 891 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithBatchNorm()
2228 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithInterDeviceTransfers()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 68 /*batch_variance=*/(replaceWithValue $x),
|
| D | legalize_tf.cc | 1989 Value batch_variance = rewriter.create<mhlo::GetTupleElementOp>( in matchAndRewrite() local
2003 op.getLoc(), batch_variance.getType(), batch_variance, in matchAndRewrite()
2053 /*reserve_space_2=*/batch_variance}); in matchAndRewrite()
2072 /*reserve_space_2=*/batch_variance, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 148 batch_norm.batch_variance); in CreateSimpleFusedBatchNorm()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 4556 TensorFlow to compute the running variance.}]>:$batch_variance,
4730 TensorFlow to compute the running variance.}]>:$batch_variance,
4780 TensorFlow to compute the running variance.}]>:$batch_variance,
18316 TF_Float32Tensor:$batch_variance,
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 21604 …ptional ...FusedBatchNormV2Attr) (y tf.Output, batch_mean tf.Output, batch_variance tf.Output, res…
24854 … optional ...FusedBatchNormAttr) (y tf.Output, batch_mean tf.Output, batch_variance tf.Output, res…
36005 …ptional ...FusedBatchNormV3Attr) (y tf.Output, batch_mean tf.Output, batch_variance tf.Output, res…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 17414 name: "batch_variance"
17765 name: "batch_variance"
17862 name: "batch_variance"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 12780 name: "batch_variance"
|
| D | ops_history.v2.pbtxt | 27176 name: "batch_variance"
27424 name: "batch_variance"
|