Home
last modified time | relevance | path

Searched refs:ExecuteSharded (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_client.h467 virtual StatusOr<std::vector<std::unique_ptr<PjRtBuffer>>> ExecuteSharded(
Dpjrt_stream_executor_client.h654 StatusOr<std::vector<std::unique_ptr<PjRtBuffer>>> ExecuteSharded(
Dpjrt_stream_executor_client.cc2057 PjRtStreamExecutorExecutable::ExecuteSharded( in ExecuteSharded() function in xla::PjRtStreamExecutorExecutable