Searched refs:WindowedInputShardReturnValue (Results 1 – 2 of 2) sorted by relevance
263 struct WindowedInputShardReturnValue { struct273 std::tuple<HloSharding, Window, WindowedInputShardReturnValue>> argument330 absl::optional<WindowedInputShardReturnValue> ReshardAsWindowedInput(
545 absl::optional<PartitionedHlo::WindowedInputShardReturnValue>557 auto update_cache = [&](WindowedInputShardReturnValue result) { in ReshardAsWindowedInput()763 return update_cache(WindowedInputShardReturnValue{ in ReshardAsWindowedInput()818 return update_cache(WindowedInputShardReturnValue{ in ReshardAsWindowedInput()