Searched defs:batch_mean (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 105 Tensor batch_mean; in VerifyTensorsClose() local
231 Tensor* batch_mean, Tensor* batch_var) { in VerifyFusedBatchNorm()
277 Tensor* batch_mean, Tensor* batch_var) { in VerifyFusedBatchNorm()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 182 Eigen::Tensor<U, 1, Eigen::RowMajor> batch_mean(depth); in operator ()() local
290 typename TTypes<U>::Vec batch_mean(batch_mean_output->vec<U>()); in operator ()() local
1304 Tensor* batch_mean = nullptr; in ComputeWithReservedSpace() local
|
| D | fused_batch_norm_ex_op_test.cc | 52 Tensor batch_mean; member
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 890 auto batch_mean = batch_norm.batch_mean; in CreateGrapplerItemWithBatchNorm() local
2227 auto batch_mean = batch_norm.batch_mean; in CreateGrapplerItemWithInterDeviceTransfers() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1496 void* batch_mean, void* batch_var, in SetBatchNormForwardArgs()
3530 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
3550 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
3572 DeviceMemory<U>* batch_mean, DeviceMemory<U>* batch_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1020 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
1039 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
|
| D | stream.cc | 351 DeviceMemory<float> *batch_mean, DeviceMemory<float> *batch_var, in ThenBatchNormalizationForward()
402 DeviceMemory<float> *batch_mean, DeviceMemory<float> *batch_var, in ThenBatchNormalizationForward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3456 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
3479 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
3504 DeviceMemory<U>* batch_mean, DeviceMemory<U>* batch_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2688 XlaOp XlaBuilder::BatchNormGrad(XlaOp operand, XlaOp scale, XlaOp batch_mean, in BatchNormGrad()
4955 const XlaOp batch_mean, const XlaOp batch_var, in BatchNormGrad()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1986 Value batch_mean = rewriter.create<mhlo::GetTupleElementOp>( in matchAndRewrite() local
|