Searched refs:input_hlo (Results 1 – 1 of 1) sorted by relevance
1817 const HloInstruction* input_hlo = hlo->operand(0); in EmitElementalDynamicSlice() local1818 const int64 rank = input_hlo->shape().rank(); in EmitElementalDynamicSlice()1835 input_hlo->shape().dimensions(i) - hlo->shape().dimensions(i); in EmitElementalDynamicSlice()1854 llvm_ir::IrArray::Index input_index(input_multi_index, input_hlo->shape(), in EmitElementalDynamicSlice()1856 return operand_to_generator.at(input_hlo)(input_index); in EmitElementalDynamicSlice()1968 const HloInstruction* input_hlo = hlo->operand(0); in EmitElementalDynamicUpdateSlice() local1972 const int64 rank = input_hlo->shape().rank(); in EmitElementalDynamicUpdateSlice()1996 input_hlo->shape().dimensions(i) - update_hlo->shape().dimensions(i); in EmitElementalDynamicUpdateSlice()2043 operand_to_generator.at(input_hlo)(index)); in EmitElementalDynamicUpdateSlice()