Home
last modified time | relevance | path

Searched defs:gather_details (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.h242 absl::Span<const Shape> shapes, std::vector<GatherDetails> gather_details, in MakeCrossHostReceiveBuffersForGather()
H A Dtfrt_cpu_pjrt_client.h202 absl::Span<const Shape> shapes, std::vector<GatherDetails> gather_details, in MakeCrossHostReceiveBuffersForGather()
H A Dpjrt_stream_executor_client.h299 std::optional<std::vector<GatherDetails>> gather_details) const { in EnqueueCrossHostReceive()
H A Dpjrt_stream_executor_client.cc1032 absl::Span<const Shape> shapes, std::vector<GatherDetails> gather_details, in MakeCrossHostReceiveBuffersForGather()