Searched refs:stack_op (Results 1 – 6 of 6) sorted by relevance
33 auto* stack_op = static_cast<StackOperator*>(stack_it->get()); in Run() local34 if (stack_op->inputs.size() > 1) { in Run()38 CHECK_EQ(stack_op->outputs.size(), 1); in Run()40 const auto& input_array = model->GetArray(stack_op->inputs[0]); in Run()51 AddMessageF("Converting trivial %s to a reshape", LogName(*stack_op)); in Run()55 reshape_op->inputs = {stack_op->inputs[0]}; in Run()56 reshape_op->outputs = stack_op->outputs; in Run()60 AvailableArrayName(*model, stack_op->outputs[0] + "_shape"); in Run()75 CHECK_EQ(stack_it->get(), stack_op); in Run()
153 auto* stack_op = new StackOperator; in Run() local154 stack_op->inputs = stack_inputs; in Run()155 stack_op->outputs = {batch_op->outputs[0]}; in Run()156 stack_op->axis = 0; in Run()157 model->operators.emplace(tail_it, stack_op); in Run()
547 const AllocatedOperand* stack_op = AllocatedOperand::New( in VerifyGapMoves() local550 block_assessments->AddDefinition(*stack_op, virtual_register); in VerifyGapMoves()
1121 auto* stack_op = tensorflow_graph->add_node(); in ConvertStackOperator() local1122 stack_op->set_op("Stack"); in ConvertStackOperator()1123 stack_op->set_name(src_op.outputs[0]); in ConvertStackOperator()1125 *stack_op->add_input() = input; in ConvertStackOperator()1127 (*stack_op->mutable_attr())["elem_type"].set_type( in ConvertStackOperator()1129 (*stack_op->mutable_attr())["axis"].set_i(src_op.axis); in ConvertStackOperator()
50 bool stack_op = (rd_is_zr || RnIsZROrSP(instr)) && in VisitAddSubImmediate() local61 if (stack_op) { in VisitAddSubImmediate()
64 bool stack_op = in VisitAddSubImmediate() local76 if (stack_op) { in VisitAddSubImmediate()