Searched refs:SourceIndexOfReshape (Results 1 – 4 of 4) sorted by relevance
123 Index SourceIndexOfReshape(const Shape& output_shape,
182 IrArray::Index IrArray::Index::SourceIndexOfReshape( in SourceIndexOfReshape() function in xla::llvm_ir::IrArray::Index309 return SourceIndexOfReshape(shape, operand_shape, builder); in SourceIndexOfBitcast()
2448 index.SourceIndexOfReshape(hlo->shape(), operand->shape(), b_)); in MakeElementGenerator()
2672 desc.scatter_indices_gen(raw_scatter_index_index.SourceIndexOfReshape( in EmitScatter()