Searched refs:spatial_dimensions_to_split (Results 1 – 1 of 1) sorted by relevance
72 std::vector<int64_t> spatial_dimensions_to_split; member131 std::vector<int64_t>* spatial_dimensions_to_split,137 absl::Span<const int64_t> spatial_dimensions_to_split,154 absl::Span<const int64_t> spatial_dimensions_to_split,206 std::vector<int64_t>* spatial_dimensions_to_split,214 absl::Span<const int64_t> spatial_dimensions_to_split,228 absl::Span<const int64_t> spatial_dimensions_to_split,511 absl::Span<const int64_t> spatial_dimensions_to_split, in HaloDuplicateWithSlice() argument514 const int64_t spatial_dim_count = spatial_dimensions_to_split.size(); in HaloDuplicateWithSlice()522 activations->shape().dimensions(spatial_dimensions_to_split[0]); in HaloDuplicateWithSlice()[all …]