Searched defs:partitioned_shape (Results 1 – 1 of 1) sorted by relevance
3015 auto partitioned_shape = in HandleDynamicSlice() local3121 const auto& partitioned_shape = partitioned_input->shape(); in HandleDynamicUpdateSlice() local3222 auto partitioned_shape = in HandleDynamicUpdateSlice() local3279 auto partitioned_shape = in HandleInfeed() local3707 auto partitioned_shape = in HandleOutfeed() local