| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | resample.py | 137 batch_mean = math_ops.reduce_mean(weights)
139 estimated_mean, batch_mean, mean_decay)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 85 typename TTypes<U>::Vec batch_mean(batch_mean_output->vec<U>()); in operator ()() local
120 batch_mean.device(d) = mean; in operator ()()
246 Tensor* batch_mean, Tensor* batch_var, Tensor* saved_mean, in operator ()()
267 f(context->eigen_device<GPUDevice>(), batch_mean->flat<U>()); in operator ()()
324 auto batch_mean_ptr = StreamExecutorUtil::AsDeviceMemory<U>(*batch_mean); in operator ()()
574 Tensor* batch_mean = nullptr; in Compute() local
576 context->allocate_output(1, scale.shape(), &batch_mean)); in Compute()
589 y, batch_mean, batch_var, saved_mean, saved_maybe_inv_var, in Compute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNorm.pbtxt | 42 name: "batch_mean"
|
| D | api_def_FusedBatchNormV2.pbtxt | 42 name: "batch_mean"
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 273 auto batch_mean = batch_norm.batch_mean; in CreateGrapplerItemWithBatchNorm() local
280 batch_mean.op(), in CreateGrapplerItemWithBatchNorm()
1561 auto batch_mean = batch_norm.batch_mean; in CreateGrapplerItemWithInterDeviceTransfers() local
1569 s.WithOpName("batch_mean1").WithDevice(kCPU1), batch_mean); in CreateGrapplerItemWithInterDeviceTransfers()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 224 DeviceMemory<float>* y, DeviceMemory<float>* batch_mean,
237 DeviceMemory<Eigen::half>* y, DeviceMemory<float>* batch_mean,
565 DeviceMemory<T>* y, DeviceMemory<U>* batch_mean,
|
| D | cuda_dnn.cc | 3185 DeviceMemory<float>* y, DeviceMemory<float>* batch_mean, in DoBatchNormalizationForward() argument
3194 epsilon, y, batch_mean, batch_var, saved_mean, saved_inv_var, in DoBatchNormalizationForward()
3206 DeviceMemory<Eigen::half>* y, DeviceMemory<float>* batch_mean, in DoBatchNormalizationForward() argument
3215 epsilon, y, batch_mean, batch_var, saved_mean, saved_inv_var, in DoBatchNormalizationForward()
3229 DeviceMemory<T>* y, DeviceMemory<U>* batch_mean, DeviceMemory<U>* batch_var, in DoBatchNormalizationForwardImpl() argument
3247 CHECK_EQ(batch_mean->is_null(), batch_var->is_null()) in DoBatchNormalizationForwardImpl()
3252 if (!batch_mean->is_null() && !batch_var->is_null()) { in DoBatchNormalizationForwardImpl()
3253 stream->ThenMemZero(batch_mean, batch_mean->size()); in DoBatchNormalizationForwardImpl()
3255 batch_mean_opaque = batch_mean->opaque(); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 215 DeviceMemory<float>* y, DeviceMemory<float>* batch_mean,
228 DeviceMemory<Eigen::half>* y, DeviceMemory<float>* batch_mean,
637 DeviceMemory<T>* y, DeviceMemory<U>* batch_mean,
|
| D | rocm_dnn.cc | 1383 void* batch_mean, void* batch_var, in SetBatchNormForwardArgs() argument
1389 k_batchnorm_op_idx, &alpha, &beta, scale, offset, batch_mean, batch_var, in SetBatchNormForwardArgs()
3003 DeviceMemory<Eigen::half>* y, DeviceMemory<float>* batch_mean, in DoBatchNormalizationForward() argument
3011 batch_mean, batch_var, saved_mean, saved_inv_var, is_training, in DoBatchNormalizationForward()
3022 DeviceMemory<float>* y, DeviceMemory<float>* batch_mean, in DoBatchNormalizationForward() argument
3030 batch_mean, batch_var, saved_mean, saved_inv_var, is_training, in DoBatchNormalizationForward()
3043 DeviceMemory<T>* y, DeviceMemory<U>* batch_mean, DeviceMemory<U>* batch_var, in DoBatchNormalizationForwardImpl() argument
3059 stream->ThenMemZero(batch_mean, batch_mean->size()); in DoBatchNormalizationForwardImpl()
3065 1.0, batch_mean->opaque(), batch_var->opaque(), epsilon, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_impl.py | 1248 y, batch_mean, batch_var, _, _ = fused_batch_norm_func(
1258 return y, batch_mean, batch_var
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 579 const XlaOp& batch_mean, const XlaOp& batch_var,
986 const XlaOp& batch_mean, const XlaOp& batch_var,
1911 const XlaOp& batch_mean, const XlaOp& batch_var,
|
| D | xla_builder.cc | 2095 const XlaOp& batch_mean, const XlaOp& batch_var, in BatchNormGrad() argument
2103 TF_ASSIGN_OR_RETURN(const Shape& batch_mean_shape, GetShape(batch_mean)); in BatchNormGrad()
2116 {operand, scale, batch_mean, batch_var, grad_output}); in BatchNormGrad()
3502 const XlaOp& batch_mean, const XlaOp& batch_var, in BatchNormGrad() argument
3505 return operand.builder()->BatchNormGrad(operand, scale, batch_mean, batch_var, in BatchNormGrad()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 997 DeviceMemory<float>* y, DeviceMemory<float>* batch_mean,
1014 DeviceMemory<Eigen::half>* y, DeviceMemory<float>* batch_mean,
|
| D | stream.h | 239 DeviceMemory<float> *y, DeviceMemory<float> *batch_mean,
260 DeviceMemory<Eigen::half> *y, DeviceMemory<float> *batch_mean,
|
| D | stream.cc | 346 DeviceMemory<float> *y, DeviceMemory<float> *batch_mean, in ThenBatchNormalizationForward() argument
357 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean, in ThenBatchNormalizationForward()
396 DeviceMemory<Eigen::half> *y, DeviceMemory<float> *batch_mean, in ThenBatchNormalizationForward() argument
407 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean, in ThenBatchNormalizationForward()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 250 | `batch_mean` | `XlaOp` | 1 dimensional array (\\(\mu\\)) |
253 The `batch_mean` and `batch_var` are moments calculated across the batch and
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 9912 …t, variance tf.Output, optional ...FusedBatchNormV2Attr) (y tf.Output, batch_mean tf.Output, batch…
10719 …put, variance tf.Output, optional ...FusedBatchNormAttr) (y tf.Output, batch_mean tf.Output, batch…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 27355 name: "batch_mean"
27428 name: "batch_mean"
27916 name: "batch_mean"
27999 name: "batch_mean"
28083 name: "batch_mean"
|
| D | ops_history.v1.pbtxt | 27726 name: "batch_mean"
27799 name: "batch_mean"
28287 name: "batch_mean"
28370 name: "batch_mean"
28454 name: "batch_mean"
|
| D | ops_history.v0.pbtxt | 12776 name: "batch_mean"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 13739 name: "batch_mean"
13987 name: "batch_mean"
|