Searched refs:element_op (Results 1 – 3 of 3) sorted by relevance
73 std::unique_ptr<Operator>& element_op = *element_op_it; in Run() local74 if (!IsElementwiseOperator(element_op->type)) { in Run()78 const std::string intermediate_name = element_op->inputs[0]; in Run()107 const std::string output_name = element_op->outputs[0]; in Run()109 AddMessageF("Swapping around operators with %s and %s", LogName(*element_op), in Run()119 AvailableArrayName(*model, element_op->outputs[0] + "_reorder"); in Run()123 element_op->inputs[0] = input_name; in Run()124 element_op->outputs[0] = new_intermediate_name; in Run()139 element_op->inputs[0] = input_name; in Run()145 model->EraseArray(element_op->outputs[0]); in Run()[all …]
213 auto element_op = builder.create<GetElementOp>( in RewriteUnregisteredTFOps() local216 new_results.push_back(element_op.out()); in RewriteUnregisteredTFOps()
653 auto element_op = dyn_cast_or_null<GetTupleElementOp>( in matchAndRewrite() local655 if (!element_op || in matchAndRewrite()656 element_op.indexAttr().getInt() != element_and_idx.index() + 1 || in matchAndRewrite()657 element_op.getOperand() != tuple_predecessor) in matchAndRewrite()