Home
last modified time | relevance | path

Searched refs:kMaxOperandsAndOutputsPerFusion (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfusion_merger_test.cc320 constexpr int64 kNumParams = kMaxOperandsAndOutputsPerFusion + 1; in TEST_F()
356 .subspan(0, kMaxOperandsAndOutputsPerFusion))); in TEST_F()
Dgpu_fusible.cc408 kMaxOperandsAndOutputsPerFusion) { in FusionWouldBeTooLarge()
417 << kMaxOperandsAndOutputsPerFusion; in FusionWouldBeTooLarge()
439 return operands.size() + num_output_buffers > kMaxOperandsAndOutputsPerFusion; in FusionWouldBeTooLarge()
Dgpu_fusible.h27 constexpr int64 kMaxOperandsAndOutputsPerFusion = 64; variable
Dmulti_output_fusion_test.cc719 ASSERT_GT(kNumParams, kMaxOperandsAndOutputsPerFusion); in TEST_F()
764 kMaxOperandsAndOutputsPerFusion) in TEST_F()
Dinstruction_fusion_test.cc470 ASSERT_GT(kNumParams, kMaxOperandsAndOutputsPerFusion); in TEST_F()
491 EXPECT_LE(instr->operand_count(), kMaxOperandsAndOutputsPerFusion) in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_fusion_test.cc61 constexpr int64 kNumParams = kMaxOperandsAndOutputsPerFusion + 1; in TEST_F()