Searched defs:FusedBatchNorm (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 92 struct FusedBatchNorm<CPUDevice, T, U, /* is_training= */ true> { struct
93 void operator()(OpKernelContext* context, const Tensor& x_input, in operator ()()
230 struct FusedBatchNorm<CPUDevice, T, U, /* is_training= */ false> { struct
231 void operator()(OpKernelContext* context, const Tensor& x_input, in operator ()()
758 struct FusedBatchNorm<GPUDevice, T, U, is_training> { struct
759 void operator()(OpKernelContext* context, const Tensor& x, in operator ()()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 92 struct FusedBatchNorm { struct
94 explicit FusedBatchNorm(int fused_batch_norm) in FusedBatchNorm() function
97 int fused_batch_norm = kMissingIndex;
|
| D | remapper_test.cc | 44 TEST_F(RemapperTest, FusedBatchNorm) { in TEST_F() argument
|
| D | auto_mixed_precision_test.cc | 502 TEST_F(AutoMixedPrecisionTest, FusedBatchNorm) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 24854 func FusedBatchNorm(scope *Scope, x tf.Output, scale tf.Output, offset tf.Output, mean tf.Output, v… func
|