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 | 205 std::vector<int> tuple_index; in IsAllReduceMovable() member
353 int tuple_index = accumulation.param_tuple_indices[0]; in IsAllReduceMovable() local
577 int tuple_index = accumulation_context.param_tuple_indices[0]; in CreateNewWhileInit() local
617 int tuple_index = accumulation_context.param_tuple_indices[0]; in CreateSinkedAllReduces() local
|
| D | hlo_matchers.h | 80 int64_t tuple_index) in HloGetTupleElementMatcher()
387 ::testing::Matcher<const HloInstruction*> operand, int64_t tuple_index) { in GetTupleElement()
|
| D | while_loop_constant_sinking.cc | 33 int64_t tuple_index) { in ReplaceUsesWhileKeepingLoopInvariance()
|
| D | hlo_module_dce_test.cc | 47 const int64_t tuple_index) { in WhileBodyHasPassThroughTupleElement()
|
| D | hlo_evaluator.cc | 304 HloInstruction* loop_body_root, int64_t tuple_index) { in PatternMatchInductionVarUpdate()
376 HloInstruction* loop_body_root, int64_t tuple_index) { in PatternMatchLoopCondVarOverride()
435 HloInstruction* param_instruction, int64_t tuple_index) { in EvaluateWhileLoopParamInitValue()
996 int64_t tuple_index = new_shape_index.front(); in EvaluateInternal() local
1499 int64_t tuple_index = visitor_shape_index_.front(); in HandleTuple() local
|
| D | hlo.proto | 130 int64 tuple_index = 13; field
|
| D | while_loop_simplifier.cc | 472 HloInstruction* while_op, const int64_t tuple_index, in TryRemoveRepeatedWhileTupleIndicesHelper()
|
| D | hlo_instructions.h | 1451 int64_t tuple_index() const { return tuple_index_; } in tuple_index() function
|
| D | layout_assignment.cc | 555 auto tuple_index = user->tuple_index(); in PropagateParameterLayoutToUsers() local
|
| D | hlo_instruction.cc | 4659 int64_t 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
|