Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dwhile_loop_all_reduce_code_motion.cc205 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 Dhlo_matchers.h80 int64_t tuple_index) in HloGetTupleElementMatcher()
387 ::testing::Matcher<const HloInstruction*> operand, int64_t tuple_index) { in GetTupleElement()
H A Dwhile_loop_constant_sinking.cc33 int64_t tuple_index) { in ReplaceUsesWhileKeepingLoopInvariance()
H A Dhlo_module_dce_test.cc47 const int64_t tuple_index) { in WhileBodyHasPassThroughTupleElement()
H A Dhlo_evaluator.cc304 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 Dhlo.proto130 int64 tuple_index = 13; field
H A Dwhile_loop_simplifier.cc472 HloInstruction* while_op, const int64_t tuple_index, in TryRemoveRepeatedWhileTupleIndicesHelper()
H A Dhlo_instructions.h1451 int64_t tuple_index() const { return tuple_index_; } in tuple_index() function
H A Dlayout_assignment.cc555 auto tuple_index = user->tuple_index(); in PropagateParameterLayoutToUsers() local
H A Dhlo_instruction.cc4659 int64_t HloInstruction::tuple_index() const { in tuple_index() function in xla::HloInstruction
/aosp_15_r20/external/python/cpython3/Objects/clinic/
Dtupleobject.c.h21 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 Dwhile_transformer_test.cc38 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 Dgenerated_variations.rs49 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 Dflatbuffer_serializer.cpp591 std::vector<uint32_t> tuple_index; in objectToFB() local
/aosp_15_r20/external/python/cpython3/Objects/
Dgenericaliasobject.c172 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/
Dtuples.hpp422 struct tuple_index {}; struct
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py12565 def tuple_index(a): function