Searched defs:spatial_dimensions_to_split (Results 1 – 1 of 1) sorted by relevance
72 std::vector<int64_t> spatial_dimensions_to_split; member511 absl::Span<const int64_t> spatial_dimensions_to_split, in HaloDuplicateWithSlice()642 std::vector<int64_t>* spatial_dimensions_to_split, bool is_backprop, in BringSpaceNextToBatch()2856 absl::Span<const int64_t> spatial_dimensions_to_split, in PerformSplitSpace()2902 absl::Span<const int64_t> spatial_dimensions_to_split, in PadAndSplitSpace()2939 std::vector<int64_t>* spatial_dimensions_to_split, bool is_backprop, in SplitSpace()3180 std::vector<int64_t> spatial_dimensions_to_split( in PropagateOnBackpropFilterConv() local3562 std::vector<int64_t> spatial_dimensions_to_split = in GetConvolutionDetails() local3717 std::vector<int64_t> spatial_dimensions_to_split = in PerformSpaceToBatchOnConvolution() local