Home
last modified time | relevance | path

Searched refs:extra_parameters (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dstable_sort_expander.cc82 std::vector<std::unique_ptr<HloInstruction>> extra_parameters; in ExpandInstruction() local
85 extra_parameters.push_back(HloInstruction::CreateParameter( in ExpandInstruction()
88 extra_parameter_ptrs.push_back(extra_parameters.back().get()); in ExpandInstruction()
89 extra_parameters.push_back(HloInstruction::CreateParameter( in ExpandInstruction()
92 extra_parameter_ptrs.push_back(extra_parameters.back().get()); in ExpandInstruction()
Dhlo_computation.h343 absl::Span<const HloInstruction* const> extra_parameters = {},
Dhlo_computation_test.cc523 std::vector<const HloInstruction*> extra_parameters{param3.get()}; in TEST_F() local
525 extra_parameters); in TEST_F()
Dhlo_computation.cc931 absl::Span<const HloInstruction* const> extra_parameters, in CloneWithReplacements() argument
998 for (const auto& instr : extra_parameters) { in CloneWithReplacements()