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.cc377 std::vector<HloInstruction*> concated_outputs; in CreateFusedComputation() local
396 concated_outputs.push_back(concated_output); in CreateFusedComputation()
400 std::vector<HloInstruction*> output_slices(concated_outputs.size() * in CreateFusedComputation()
402 for (size_t i = 0; i < concated_outputs.size(); ++i) { in CreateFusedComputation()
403 auto concated_output = concated_outputs[i]; in CreateFusedComputation()
411 output_slices[concated_outputs.size() * j + i], in CreateFusedComputation()