HomeSort by relevance Sort by last modified time
    Searched refs:FusedBatchNorm (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/
fused_batchnorm_reserve_space_test.cc 59 // This tests check that the implementation outputs from FusedBatchnorm
65 // lowering for FusedBatchnorm training to match the new implementation defined
69 using ::tensorflow::ops::FusedBatchNorm;
100 FusedBatchNorm fused_batch_norm_tf(
102 scale, offset, mean, variance, FusedBatchNorm::Attrs{}.IsTraining(true));
103 FusedBatchNorm fused_batch_norm_xla(
105 scale, offset, mean, variance, FusedBatchNorm::Attrs{}.IsTraining(true));
  /external/tensorflow/tensorflow/core/grappler/optimizers/
remapper_test.cc 39 TEST_F(RemapperTest, FusedBatchNorm) {
47 ops::FusedBatchNorm::Attrs attr;
49 ops::FusedBatchNorm bn(s.WithOpName("batch_norm"), x, scale, offset, mean,
80 ops::FusedBatchNorm::Attrs attr;
83 ops::FusedBatchNorm bn(s.WithOpName("batch_norm").WithDevice("/device:GPU:0"),
252 ops::FusedBatchNorm::Attrs attrs;
254 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale,
296 EXPECT_EQ("FusedBatchNorm", fused_ops[0]);
329 ops::FusedBatchNorm::Attrs attrs;
331 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale
    [all...]
remapper.cc 62 // FusedBatchNorm that can be replaced with a cheaper set of primitives.
63 struct FusedBatchNorm {
64 FusedBatchNorm() = default;
65 explicit FusedBatchNorm(const NodeDef* fused_batch_norm)
105 // Conv2D node followed by a FusedBatchNorm.
117 // Conv2D node followed by a FusedBatchNorm and Relu.
347 // Root of the pattern must be a FusedBatchNorm or a FusedBatchNormV2.
367 // Input to the FusedBatchNorm must be a Conv2D.
381 // We successfully found a Conv2D+FusedBatchNorm pattern.
410 // We successfully found a Conv2D+FusedBatchNorm+Relu pattern
    [all...]
  /external/tensorflow/tensorflow/core/profiler/g3doc/
profile_memory.md 28 FusedBatchNorm 10517.52MB (64.01%, 8.74%)
  /external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.cc 41 struct FusedBatchNorm;
66 struct FusedBatchNorm<CPUDevice, T, U> {
76 errors::Internal("The CPU implementation of FusedBatchNorm "
241 struct FusedBatchNorm<GPUDevice, T, U> {
256 VLOG(2) << "FusedBatchNorm:"
587 functor::FusedBatchNorm<Device, T, U>()(
711 Name("FusedBatchNorm").Device(DEVICE_CPU).TypeConstraint<float>("T"),
    [all...]
conv_ops_test.cc 663 ops::FusedBatchNorm::Attrs attr;
666 ops::FusedBatchNorm with_fused_batch_norm = ops::FusedBatchNorm(
690 ops::FusedBatchNorm::Attrs attr;
693 ops::FusedBatchNorm with_fused_batch_norm = ops::FusedBatchNorm(
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
virtual_scheduler_test.cc 269 auto batch_norm = ops::FusedBatchNorm(
271 ops::FusedBatchNorm::IsTraining(true).Epsilon(0.1f));
1543 // Create a FusedBatchNorm op that has multiple output ports.
1547 // Create a FusedBatchNorm op that has multiple output ports.
1557 auto batch_norm = ops::FusedBatchNorm(
1559 ops::FusedBatchNorm::IsTraining(true).Epsilon(0.1f));
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 307 milliseconds