Searched defs:partitioned_shape (Results 1 – 1 of 1) sorted by relevance
2307 auto partitioned_shape = in HandleDynamicSlice() local2375 const auto& partitioned_shape = partitioned_input->shape(); in HandleDynamicUpdateSlice() local2467 auto partitioned_shape = in HandleDynamicUpdateSlice() local2524 auto partitioned_shape = in HandleInfeed() local2926 auto partitioned_shape = in HandleOutfeed() local