Searched defs:FusedBatchNorm (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-043.md | 3 ### CVE Number
6 ### Impact
42 ### Patches
50 ### For more information
56 ### Attribution
|
| D | tfsa-2021-072.md | 3 ### CVE Number
6 ### Impact
62 ### Patches
70 ### For more information
76 ### Attribution
|
| /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 ()()
233 struct FusedBatchNorm<CPUDevice, T, U, /* is_training= */ false> { struct
234 void operator()(OpKernelContext* context, const Tensor& x_input, in operator ()()
768 struct FusedBatchNorm<GPUDevice, T, U, is_training> { struct
769 void operator()(OpKernelContext* context, const Tensor& x, in operator ()()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 106 struct FusedBatchNorm { struct
108 explicit FusedBatchNorm(int fused_batch_norm) in FusedBatchNorm() argument
111 int fused_batch_norm = kMissingIndex;
|
| D | auto_mixed_precision_test.cc | 559 TEST_F(AutoMixedPrecisionTest, FusedBatchNorm) { in TEST_F() argument
|
| D | remapper_test.cc | 47 TEST_F(RemapperTest, FusedBatchNorm) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 17704 func FusedBatchNorm(scope *Scope, x tf.Output, scale tf.Output, offset tf.Output, mean tf.Output, v… func
|