Searched refs:CastToShape (Results 1 – 3 of 3) sorted by relevance
266 IrArray CastToShape(const Shape& new_shape, llvm::IRBuilder<>* b) const;
521 IrArray IrArray::CastToShape(const Shape& new_shape, in CastToShape() function in xla::llvm_ir::IrArray
4109 output_array.CastToShape(output_reduced_shape, &b_) in EmitTileElementForCopy()4874 param_arrays[id].CastToShape(reduced_shape, &b_)); in EmitHlo021Tile()