Home
last modified time | relevance | path

Searched defs:tuple_index (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_all_reduce_code_motion.cc90 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
Dhlo_matchers.h77 int64 tuple_index) in HloGetTupleElementMatcher()
336 ::testing::Matcher<const HloInstruction*> operand, int64 tuple_index) { in GetTupleElement()
Dwhile_loop_constant_sinking.cc30 HloInstruction* while_body_root, int64 tuple_index) { in ReplaceUsesWhileKeepingLoopInvariance()
Dhlo_module_dce_test.cc48 const int64 tuple_index) { in WhileBodyHasPassThroughTupleElement()
Dtuple_points_to_analysis_test.cc787 HloInstruction* fusion_param, int64 tuple_index) { in GetFusionParameterUsersAt()
801 int64 tuple_index) { in GetUniqueFusionParameterUserAt()
Dhlo.proto75 int64 tuple_index = 13; field
Dwhile_loop_simplifier.cc335 HloInstruction* while_op, const int64 tuple_index, in TryRemoveRepeatedWhileTupleIndicesHelper()
Dhlo_instructions.h1101 int64 tuple_index() const { return tuple_index_; } in tuple_index() function
Dhlo_instruction.cc4086 int64 HloInstruction::tuple_index() const { in tuple_index() function in xla::HloInstruction
/external/python/cpython3/Objects/clinic/
Dtupleobject.c.h21 tuple_index(PyTupleObject *self, PyObject *const *args, Py_ssize_t nargs) in tuple_index() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc38 const int64 tuple_index, const int64 limit) { in BuildConditionComputation()
/external/python/cpython3/Objects/
Dgenericaliasobject.c179 tuple_index(PyObject *self, Py_ssize_t len, PyObject *item) in tuple_index() function