Searched defs:FusedBatchNorm (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 66 struct FusedBatchNorm<CPUDevice, T, U> { struct
67 void operator()(OpKernelContext* context, const Tensor& x_input, in operator ()()
241 struct FusedBatchNorm<GPUDevice, T, U> { struct
242 void operator()(OpKernelContext* context, const Tensor& x, in operator ()()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 63 struct FusedBatchNorm { struct
65 explicit FusedBatchNorm(const NodeDef* fused_batch_norm) in FusedBatchNorm() argument
68 const NodeDef* fused_batch_norm = nullptr;
|
| D | remapper_test.cc | 39 TEST_F(RemapperTest, FusedBatchNorm) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 10719 func FusedBatchNorm(scope *Scope, x tf.Output, scale tf.Output, offset tf.Output, mean tf.Output, v… func
|