Searched refs:split_spatial_dimensions (Results 1 – 1 of 1) sorted by relevance
2394 std::vector<int64_t> split_spatial_dimensions( in BatchToSpace() local2396 absl::c_iota(split_spatial_dimensions, space_dim); in BatchToSpace()2400 split_spatial_dimensions)); in BatchToSpace()2405 new_dimensions.erase(new_dimensions.begin() + split_spatial_dimensions[0], in BatchToSpace()2406 new_dimensions.begin() + split_spatial_dimensions[0] + in BatchToSpace()2409 for (auto spatial_dimension : split_spatial_dimensions) { in BatchToSpace()2424 for (auto spatial_dimension : split_spatial_dimensions) { in BatchToSpace()