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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc 139 auto* batchnorm = builder.AddInstruction(HloInstruction::CreateCustomCall( local
146 module->AddEntryComputation(builder.Build(batchnorm));
162 // The first operand to batchnorm should have the same layout as the
164 EXPECT_TRUE(LayoutUtil::Equal(batchnorm->operand(0)->shape().layout(),
165 batchnorm->shape().layout()))
166 << batchnorm->ToString();
207 auto* batchnorm = builder.AddInstruction(HloInstruction::CreateCustomCall( local
213 module->AddEntryComputation(builder.Build(batchnorm));
231 // The first operand to batchnorm should have the same layout as the
234 LayoutUtil::Equal(batchnorm->operand(0)->shape().layout()
283 auto* batchnorm = local
    [all...]
  /external/tensorflow/tensorflow/core/graph/
graph_constructor_test.cc 1099 name: "batchnorm"
    [all...]

Completed in 212 milliseconds