Home
last modified time | relevance | path

Searched refs:user_index (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_allocator_retry_test.cc79 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/
Dhlo_dataflow_analysis.h60 const ShapeIndex& user_index)>;
162 const ShapeIndex& user_index) const;
Dhlo_dataflow_analysis.cc1263 HloInstruction* user, const ShapeIndex& user_index) const { in CanShareOperandBufferWithUser()
1272 ShapeUtil::GetSubshape(user->shape(), user_index); in CanShareOperandBufferWithUser()
1283 if (operand_and_output_index.second != user_index) { in CanShareOperandBufferWithUser()
1295 can_share_buffer_(user, operand, user_index)) { in CanShareOperandBufferWithUser()
1360 CHECK(!user_index.empty()); in CanShareOperandBufferWithUser()
1363 return operand_indices.size() == 1 && user_index[0] == operand_indices[0]; in CanShareOperandBufferWithUser()
Dhlo_rematerialization.cc471 absl::optional<int64> user_index = in GetUsers() local
478 ItemUse{user_item, static_cast<int>(op_idx), user_index})) { in GetUsers()
480 ItemUse{user_item, static_cast<int>(op_idx), user_index}); in GetUsers()
Dheap_simulator_test.cc502 const ShapeIndex& user_index) -> absl::optional<bool> { in TEST_F() argument
576 const ShapeIndex& user_index) -> absl::optional<bool> { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc187 const ShapeIndex& user_index) { in CanShareBufferHint() argument
197 return user_index.size() == 1 && user_index[0] == 0; in CanShareBufferHint()
/external/llvm-project/polly/lib/External/ppcg/
Dgpu.h218 void *user_index,
/external/kernel-headers/original/uapi/rdma/
Dmlx5-abi.h403 __u32 user_index; member
/external/tensorflow/tensorflow/core/kernels/
Ddecode_proto_op.cc201 FieldInfo(const FieldDescriptor* field_desc, int user_index, in FieldInfo()
203 : output_index(user_index), default_value(def_value) { in FieldInfo()
/external/llvm-project/polly/lib/External/pet/include/
Dpet.h410 void *user), void *user_index,
/external/selinux/libsepol/src/
Dpolicydb.c1057 static int user_index(hashtab_key_t key, hashtab_datum_t datum, void *datap) in user_index() function
1116 common_index, class_index, role_index, type_index, user_index,
/external/python/cpython2/Tools/scripts/
Dtexi2html.py1548 def user_index(self, name, args): member in TexinfoParser
/external/python/cpython3/Tools/scripts/
Dtexi2html.py1546 def user_index(self, name, args): member in TexinfoParser