Searched refs:SetLayoutForMlir (Results 1 – 3 of 3) sorted by relevance
65 static void SetLayoutForMlir(mlir::Operation* op, const Shape& shape);
856 SetLayoutForMlir(op, instruction->shape()); in ImportInstruction()990 void HloFunctionImporter::SetLayoutForMlir(mlir::Operation* op, in SetLayoutForMlir() function in xla::HloFunctionImporter
650 HloFunctionImporter::SetLayoutForMlir(load, in EmitUsingElementalIrEmitter()659 HloFunctionImporter::SetLayoutForMlir( in EmitUsingElementalIrEmitter()663 HloFunctionImporter::SetLayoutForMlir(fused_copy, output_shape); in EmitUsingElementalIrEmitter()678 HloFunctionImporter::SetLayoutForMlir(fused_reduce, shape); in EmitUsingElementalIrEmitter()724 HloFunctionImporter::SetLayoutForMlir(new_op, output_shape); in EmitUsingElementalIrEmitter()