/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A 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
|
H A D | hlo_matchers.h | 80 int64_t tuple_index) in HloGetTupleElementMatcher() 387 ::testing::Matcher<const HloInstruction*> operand, int64_t tuple_index) { in GetTupleElement()
|
H A D | while_loop_constant_sinking.cc | 33 int64_t tuple_index) { in ReplaceUsesWhileKeepingLoopInvariance()
|
H A D | hlo_module_dce_test.cc | 47 const int64_t tuple_index) { in WhileBodyHasPassThroughTupleElement()
|
H A 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
|
H A D | hlo.proto | 130 int64 tuple_index = 13; field
|
H A D | while_loop_simplifier.cc | 472 HloInstruction* while_op, const int64_t tuple_index, in TryRemoveRepeatedWhileTupleIndicesHelper()
|
H A D | hlo_instructions.h | 1451 int64_t tuple_index() const { return tuple_index_; } in tuple_index() function
|
H A D | layout_assignment.cc | 555 auto tuple_index = user->tuple_index(); in PropagateParameterLayoutToUsers() local
|
H A D | hlo_instruction.cc | 4659 int64_t HloInstruction::tuple_index() const { in tuple_index() function in xla::HloInstruction
|
/aosp_15_r20/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
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | while_transformer_test.cc | 38 const int64_t tuple_index, const int64_t limit) { in BuildConditionComputation()
|
/aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/read-fonts-0.15.6/generated/ |
H A D | generated_variations.rs | 49 let tuple_index: TupleIndex = cursor.read()?; in read_with_args() localVariable 91 pub fn tuple_index(&self) -> TupleIndex { in tuple_index() method
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
H A D | flatbuffer_serializer.cpp | 591 std::vector<uint32_t> tuple_index; in objectToFB() local
|
/aosp_15_r20/external/python/cpython3/Objects/ |
D | genericaliasobject.c | 172 tuple_index(PyObject *self, Py_ssize_t len, PyObject *item) in tuple_index() function
|
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/spirit/include/boost/spirit/home/classic/phoenix/ |
D | tuples.hpp | 422 struct tuple_index {}; struct
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_jit.py | 12565 def tuple_index(a): function
|