Searched refs:SourceIndexOfReshape (Results 1 – 4 of 4) sorted by relevance
130 IrArray::Index IrArray::Index::SourceIndexOfReshape( in SourceIndexOfReshape() function in xla::llvm_ir::IrArray::Index225 return SourceIndexOfReshape(shape, operand_shape, builder); in SourceIndexOfBitcast()
125 Index SourceIndexOfReshape(const Shape& output_shape,
2405 index.SourceIndexOfReshape(hlo->shape(), operand->shape(), b_)); in MakeElementGenerator()
1150 scatter_indices_gen(raw_scatter_index_index.SourceIndexOfReshape( in EmitScatter()