Searched refs:user_index (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_allocator_retry_test.cc | 79 void WaitTurn(int user_index) { in WaitTurn() argument
83 while (next_turn_ != user_index && wait_cycles++ < 10) { in WaitTurn()
86 if (next_turn_ == user_index) { in WaitTurn()
93 void Done(int user_index) { in Done() argument
95 done_[user_index] = true; in Done()
96 if (next_turn_ == user_index) { in Done()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 170 const ShapeIndex& user_index) { in CanShareBufferHint() argument
175 (user_index.size() == 1 && in CanShareBufferHint()
176 user->operand(user_index[0]) == operand); in CanShareBufferHint()
186 return user_index.size() == 1 && user_index[0] == 0; in CanShareBufferHint()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis.h | 76 const ShapeIndex& user_index)>;
176 const ShapeIndex& user_index) const;
|
| D | hlo_dataflow_analysis.cc | 1845 HloInstruction* user, const ShapeIndex& user_index) const { in CanShareOperandBufferWithUser()
1855 ShapeUtil::GetSubshape(user->shape(), user_index); in CanShareOperandBufferWithUser()
1866 fusion_param, user->fused_expression_root(), user_index); in CanShareOperandBufferWithUser()
1876 if (operand_and_output_index.second != user_index) { in CanShareOperandBufferWithUser()
1891 can_share_buffer_(user, operand, user_index)) { in CanShareOperandBufferWithUser()
1971 CHECK(!user_index.empty()); in CanShareOperandBufferWithUser()
1974 return operand_indices.size() == 1 && user_index[0] == operand_indices[0]; in CanShareOperandBufferWithUser()
|
| D | hlo_rematerialization.cc | 475 std::optional<int64_t> user_index = in GetUsers() local
482 ItemUse{user_item, static_cast<int>(op_idx), user_index})) { in GetUsers()
484 ItemUse{user_item, static_cast<int>(op_idx), user_index}); in GetUsers()
|
| D | heap_simulator_test.cc | 502 const ShapeIndex& user_index) -> std::optional<bool> { in TEST_F() argument
506 ShapeUtil::GetSubshape(instr->shape(), user_index)); in TEST_F()
576 const ShapeIndex& user_index) -> std::optional<bool> { in TEST_F() argument
|
| /external/kernel-headers/original/uapi/rdma/ |
| D | mlx5-abi.h | 420 __u32 user_index; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | decode_proto_op.cc | 201 FieldInfo(const FieldDescriptor* field_desc, int user_index, in FieldInfo()
203 : output_index(user_index), default_value(def_value) { in FieldInfo()
|
| /external/selinux/libsepol/src/ |
| D | policydb.c | 1073 static int user_index(hashtab_key_t key, hashtab_datum_t datum, void *datap) in user_index() function
1131 common_index, class_index, role_index, type_index, user_index,
|
| /external/python/cpython2/Tools/scripts/ |
| D | texi2html.py | 1548 def user_index(self, name, args): member in TexinfoParser
|
| /external/python/cpython3/Tools/scripts/ |
| D | texi2html.py | 1546 def user_index(self, name, args): member in TexinfoParser
|