Searched defs:async_group_id (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 195 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()
|
| D | hlo.proto | 334 int64 async_group_id = 78; field
|
| D | hlo_instructions.cc | 246 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()
|
| D | hlo_instructions.h | 237 std::optional<int64_t> async_group_id() const { return async_group_id_; } in async_group_id() function
|
| D | hlo_parser.cc | 1618 optional<int64_t> async_group_id; in CreateInstruction() local
|