Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion.cc392 std::vector<HloInstruction*> concated_outputs; in CreateFusedComputation() local
413 concated_outputs.push_back(concated_output); in CreateFusedComputation()
417 std::vector<HloInstruction*> output_slices(concated_outputs.size() * in CreateFusedComputation()
419 for (size_t i = 0; i < concated_outputs.size(); ++i) { in CreateFusedComputation()
420 HloInstruction* concated_output = concated_outputs[i]; in CreateFusedComputation()
429 output_slices[concated_outputs.size() * j + i], in CreateFusedComputation()