Searched defs:tuple_index (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_all_reduce_code_motion.cc | 90 std::vector<int> tuple_index; in IsAllReduceMovable() member
212 int tuple_index = accumulation.param_tuple_indices[0]; in IsAllReduceMovable() local
369 int tuple_index = accumulation_context.param_tuple_indices[0]; in CreateNewWhileInit() local
409 int tuple_index = accumulation_context.param_tuple_indices[0]; in CreateSinkedAllReduces() local
|
| D | hlo_matchers.h | 80 int64_t tuple_index) in HloGetTupleElementMatcher()
370 ::testing::Matcher<const HloInstruction*> operand, int64_t tuple_index) { in GetTupleElement()
|
| D | while_loop_constant_sinking.cc | 30 HloInstruction* while_body_root, int64_t tuple_index) { in ReplaceUsesWhileKeepingLoopInvariance()
|
| D | hlo_module_dce_test.cc | 48 const int64_t tuple_index) { in WhileBodyHasPassThroughTupleElement()
|
| D | tuple_points_to_analysis_test.cc | 787 HloInstruction* fusion_param, int64_t tuple_index) { in GetFusionParameterUsersAt()
801 int64_t tuple_index) { in GetUniqueFusionParameterUserAt()
|
| D | hlo.proto | 110 int64 tuple_index = 13; field
|
| D | while_loop_simplifier.cc | 368 HloInstruction* while_op, const int64_t tuple_index, in TryRemoveRepeatedWhileTupleIndicesHelper()
|
| D | hlo_instructions.h | 1169 int64 tuple_index() const { return tuple_index_; } in tuple_index() function
|
| D | layout_assignment.cc | 491 auto tuple_index = user->tuple_index(); in PropagateParameterLayoutToUsers() local
|
| D | hlo_instruction.cc | 4384 int64 HloInstruction::tuple_index() const { in tuple_index() function in xla::HloInstruction
|
| /external/python/cpython3/Objects/clinic/ |
| D | tupleobject.c.h | 21 tuple_index(PyTupleObject *self, PyObject *const *args, Py_ssize_t nargs) in tuple_index() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | while_transformer_test.cc | 38 const int64_t tuple_index, const int64_t limit) { in BuildConditionComputation()
|
| /external/python/cpython3/Objects/ |
| D | genericaliasobject.c | 180 tuple_index(PyObject *self, Py_ssize_t len, PyObject *item) in tuple_index() function
|