Searched defs:dims_to_remove (Results 1 – 2 of 2) sorted by relevance
532 exec_aten::SizesType dims_to_remove = 0; in get_squeeze_copy_dims_out_target_size() local
1351 absl::Span<const int64_t> dims_to_remove) { in RemoveShapeDimensions()