Searched refs:input_shape_info (Results 1 – 1 of 1) sorted by relevance
249 const ShapeInfo& input_shape_info, const ShapeInfo& filter_shape_info, in CreateNaiveMlirConv() argument252 CHECK(input_shape_info.element_type == builder.getF16Type()); in CreateNaiveMlirConv()322 mlir::AffineMap(input_shape_info.affine_map) in CreateNaiveMlirConv()535 ShapeInfo input_shape_info = in EmitConvolutionForwardAsMlir() local556 mlir::MemRefType::get(input_shape_info.physical_dimensions, in EmitConvolutionForwardAsMlir()557 input_shape_info.element_type, {}), in EmitConvolutionForwardAsMlir()575 CreateNaiveMlirConv(input, filter, output, input_shape_info, in EmitConvolutionForwardAsMlir()