Searched refs:input_hlo (Results 1 – 1 of 1) sorted by relevance
1864 const HloInstruction* input_hlo = hlo->operand(0); in EmitElementalDynamicSlice() local1865 const int64 rank = input_hlo->shape().rank(); in EmitElementalDynamicSlice()1882 input_hlo->shape().dimensions(i) - hlo->shape().dimensions(i); in EmitElementalDynamicSlice()1901 llvm_ir::IrArray::Index input_index(input_multi_index, input_hlo->shape(), in EmitElementalDynamicSlice()1903 return operand_to_generator.at(input_hlo)(input_index); in EmitElementalDynamicSlice()2027 const HloInstruction* input_hlo = hlo->operand(0); in EmitElementalDynamicUpdateSlice() local2031 const int64 rank = input_hlo->shape().rank(); in EmitElementalDynamicUpdateSlice()2055 input_hlo->shape().dimensions(i) - update_hlo->shape().dimensions(i); in EmitElementalDynamicUpdateSlice()2102 operand_to_generator.at(input_hlo)(index)); in EmitElementalDynamicUpdateSlice()