Searched refs:nchw_dimensions (Results 1 – 1 of 1) sorted by relevance
55 std::vector<int64_t> nchw_dimensions; member85 shape_info.nchw_dimensions.push_back(shape.dimensions(dim)); in GetShapeInfo()259 CreateNestedSimpleLoops(output_shape_info.nchw_dimensions, builder); in CreateNaiveMlirConv()275 absl::MakeSpan(filter_shape_info.nchw_dimensions).subspan(1), builder); in CreateNaiveMlirConv()