Home
last modified time | relevance | path

Searched refs:tuple_handle (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1644 absl::optional<TupleHandle> tuple_handle; in MakeExecutionInputsAndWaitForEvents() local
1646 TF_ASSIGN_OR_RETURN(tuple_handle, in MakeExecutionInputsAndWaitForEvents()
1649 events.insert(tuple_handle->event.get()); in MakeExecutionInputsAndWaitForEvents()
1650 execution_inputs.emplace_back(std::move(tuple_handle->execution_input)); in MakeExecutionInputsAndWaitForEvents()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc254 std::unique_ptr<tpu_driver::BufferHandle> tuple_handle = in MakeTuple() local
258 client->driver(), std::move(tuple_handle), std::move(child_events), in MakeTuple()
/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto534 GlobalDataHandle tuple_handle = 2; field
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc210 allocation_tracker_.DeconstructTuple(arg->tuple_handle())); in DeconstructTuple()