Searched refs:apply_layout (Results 1 – 1 of 1) sorted by relevance
1134 auto apply_layout = [](const Shape& shape, mlir::ArrayAttr layout_attrib) { in EmitConvolutionThunkFromMlir() local1147 apply_layout(TypeToShape(input.op->getOperand(0).getType()), in EmitConvolutionThunkFromMlir()1150 apply_layout(TypeToShape(input.op->getOperand(1).getType()), in EmitConvolutionThunkFromMlir()1152 descriptor.result_shape = apply_layout(TypeToShape(conv_result.getType()), in EmitConvolutionThunkFromMlir()