Searched defs:partitioned_shape (Results 1 – 1 of 1) sorted by relevance
2269 auto partitioned_shape = in HandleDynamicSlice() local2345 const auto& partitioned_shape = partitioned_input->shape(); in HandleDynamicUpdateSlice() local2441 auto partitioned_shape = in HandleDynamicUpdateSlice() local2498 auto partitioned_shape = in HandleInfeed() local2922 auto partitioned_shape = in HandleOutfeed() local