Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc129 std::vector<HloInstruction*> expanded_index_components; in ExpandIndexVectorIntoOperandSpace() local
139 expanded_index_components.push_back(component_to_concat); in ExpandIndexVectorIntoOperandSpace()
141 expanded_index_components.push_back(zero); in ExpandIndexVectorIntoOperandSpace()
145 return MakeConcatHlo(expanded_index_components, /*dimension=*/0); in ExpandIndexVectorIntoOperandSpace()
Dscatter_expander.cc150 std::vector<HloInstruction*> expanded_index_components; in ExpandIndexVectorIntoOperandSpace() local
162 expanded_index_components.push_back(component_to_concat); in ExpandIndexVectorIntoOperandSpace()
164 expanded_index_components.push_back(zero); in ExpandIndexVectorIntoOperandSpace()
168 return MakeConcatHlo(expanded_index_components, /*dimension=*/0); in ExpandIndexVectorIntoOperandSpace()