| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 86 typename TTypes<U>::Vec batch_var(batch_var_output->vec<U>()); in operator ()() local
131 batch_var.device(d) = variance * rest_size_adjust; in operator ()()
246 Tensor* batch_mean, Tensor* batch_var, Tensor* saved_mean, in operator ()()
268 f(context->eigen_device<GPUDevice>(), batch_var->flat<U>()); in operator ()()
326 auto batch_var_ptr = StreamExecutorUtil::AsDeviceMemory<U>(*batch_var); in operator ()()
577 Tensor* batch_var = nullptr; in Compute() local
579 context->allocate_output(2, scale.shape(), &batch_var)); in Compute()
589 y, batch_mean, batch_var, saved_mean, saved_maybe_inv_var, in Compute()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 274 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithBatchNorm() local
277 auto z2 = ops::Add(s.WithOpName("z2"), batch_var, batch_var); in CreateGrapplerItemWithBatchNorm()
278 auto z3 = ops::Add(s.WithOpName("z3"), batch_var, batch_var); in CreateGrapplerItemWithBatchNorm()
285 auto z4 = ops::NoOp(s.WithControlDependencies(batch_var).WithOpName("z4")); in CreateGrapplerItemWithBatchNorm()
1562 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithInterDeviceTransfers() local
1571 ops::Identity(s.WithOpName("batch_var1").WithDevice(kCPU1), batch_var); in CreateGrapplerItemWithInterDeviceTransfers()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 225 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
238 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
566 DeviceMemory<U>* batch_var, DeviceMemory<U>* saved_mean,
|
| D | cuda_dnn.cc | 3186 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward() argument
3194 epsilon, y, batch_mean, batch_var, saved_mean, saved_inv_var, in DoBatchNormalizationForward()
3207 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_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()
3254 stream->ThenMemZero(batch_var, batch_var->size()); in DoBatchNormalizationForwardImpl()
3256 batch_var_opaque = batch_var->opaque(); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 216 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
229 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
638 DeviceMemory<U>* batch_var, DeviceMemory<U>* saved_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()
3004 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward() argument
3011 batch_mean, batch_var, saved_mean, saved_inv_var, is_training, in DoBatchNormalizationForward()
3023 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_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
3060 stream->ThenMemZero(batch_var, batch_var->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
2104 TF_ASSIGN_OR_RETURN(const Shape& batch_var_shape, GetShape(batch_var)); 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 | 998 DeviceMemory<float>* batch_var, DeviceMemory<float>* reserve_space_1,
1015 DeviceMemory<float>* batch_var, DeviceMemory<float>* reserve_space_1,
|
| D | stream.h | 240 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean,
261 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean,
|
| D | stream.cc | 347 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean, in ThenBatchNormalizationForward() argument
357 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean, in ThenBatchNormalizationForward()
397 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_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 | 251 | `batch_var` | `XlaOp` | 1 dimensional array (\\(\sigma^2\\)) |
253 The `batch_mean` and `batch_var` are moments calculated across the batch and
|