Searched refs:TpuEvent (Results 1 – 5 of 5) sorted by relevance
34 typedef struct TpuEvent TpuEvent; typedef45 TpuEvent* event;51 TpuEvent* event;56 TpuEvent* event;119 int32_t num_replicas, int32_t eventc, struct TpuEvent** eventv);126 struct TpuEvent** eventv);138 int32_t eventc, struct TpuEvent** eventv);144 typedef struct TpuEvent*(PrototypeTpuDriver_UnloadProgram)(147 struct TpuEvent** eventv);149 typedef struct TpuEvent*(PrototypeTpuDriver_ExecuteProgram)([all …]
77 TpuEvent* compile_events[] = {cph->event}; in main()105 TpuEvent* allocate_buf_a_events[] = {buf_a_handle->event}; in main()107 struct TpuEvent* transfer_ev1 = in main()110 TpuEvent* allocate_buf_b_events[] = {buf_a_handle->event}; in main()112 struct TpuEvent* transfer_ev2 = in main()120 TpuEvent* transfer_events[] = {transfer_ev1, transfer_ev2}; in main()121 struct TpuEvent* execute_event = in main()129 TpuEvent* execute_events[] = {execute_event}; in main()130 struct TpuEvent* transfer_sum_event = in main()142 struct TpuEvent* unload_program_event = driver_fn.TpuDriver_UnloadProgram( in main()[all …]
25 class TpuEvent : public ::stream_executor::internal::EventInterface {27 explicit TpuEvent(SE_Event* event) : event_(event) {} in TpuEvent() function28 ~TpuEvent() override { in ~TpuEvent()
190 auto ptr = absl::make_unique<TpuEvent>(tpu_event); in CreateEventImplementation()
64 explicit DirectEvent(::TpuDriverFn* driver_fn, ::TpuEvent* event) in DirectEvent()110 ::TpuEvent* event_;474 ::TpuEvent** MakeEventArray(absl::Span<Event* const> wait_for) { in MakeEventArray()476 ::TpuEvent** ret = new ::TpuEvent*[wait_for.size()]; in MakeEventArray()