Home
last modified time | relevance | path

Searched refs:instruction_vector (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dflatbuffer_serializer.cpp248 std::vector<mobile::serialization::Instruction> instruction_vector; in functionToFB() local
249 instruction_vector.reserve(code.instructions_.size()); in functionToFB()
251 instruction_vector.emplace_back(inst.op, inst.N, inst.X); in functionToFB()
336 &instruction_vector, in functionToFB()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmemory_space_assignment.cc1636 std::vector<const HloInstruction*> instruction_vector; in CreateAllocationValuesFromColocatedIntervals() local
1637 instruction_vector.push_back(allocation_value.defining_instruction()); in CreateAllocationValuesFromColocatedIntervals()
1639 instruction_vector.push_back(use.hlo_use.instruction); in CreateAllocationValuesFromColocatedIntervals()
1641 return instruction_vector; in CreateAllocationValuesFromColocatedIntervals()