Home
last modified time | relevance | path

Searched defs:async_group_id (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction.cc195 std::optional<int64_t> async_group_id; in CreateFromProto() local
211 std::optional<int64_t> async_group_id; in CreateFromProto() local
226 std::optional<int64_t> async_group_id; in CreateFromProto() local
1189 HloComputation* async_computation, std::optional<int64_t> async_group_id, in CreateAsyncStart()
1198 HloComputation* async_computation, std::optional<int64_t> async_group_id, in CreateAsyncUpdate()
1207 HloComputation* async_computation, std::optional<int64_t> async_group_id, in CreateAsyncDone()
4862 std::optional<int64_t> HloInstruction::async_group_id() const { in async_group_id() function in xla::HloInstruction
4866 void HloInstruction::set_async_group_id(std::optional<int64_t> async_group_id) { in set_async_group_id()
H A Dhlo.proto334 int64 async_group_id = 78; field
H A Dhlo_instructions.cc246 HloComputation* async_computation, std::optional<int64_t> async_group_id, in HloAsyncInstruction()
264 HloComputation* async_computation, std::optional<int64_t> async_group_id, in HloAsyncInstruction()
351 std::optional<int64_t> async_group_id) { in set_async_group_id()
H A Dhlo_instructions.h237 std::optional<int64_t> async_group_id() const { return async_group_id_; } in async_group_id() function
H A Dhlo_parser.cc1618 optional<int64_t> async_group_id; in CreateInstruction() local