Searched refs:CastToShape (Results 1 – 3 of 3) sorted by relevance
259 IrArray CastToShape(const Shape& new_shape, llvm::IRBuilder<>* b) const;
411 IrArray IrArray::CastToShape(const Shape& new_shape, in CastToShape() function in xla::llvm_ir::IrArray
2291 param_arrays[id].CastToShape((*param_reduced_shapes)[id], &b_)); in ConstructInputReducedShapeAndCastInputIrArrayToShape()2472 output_array.CastToShape(output_reduced_shape, &b_) in EmitTileElementForCopy()