Searched refs:expand_op (Results 1 – 2 of 2) sorted by relevance
36 ExpandDimsOperator* expand_op = in Run() local38 CHECK_EQ(expand_op->inputs.size(), 2); in Run()39 CHECK_EQ(expand_op->outputs.size(), 1); in Run()41 const auto& input_array = model->GetArray(expand_op->inputs[0]); in Run()47 const auto& axis_array = model->GetArray(expand_op->inputs[1]); in Run()75 reshape_op->inputs.push_back(expand_op->inputs[0]); in Run()76 reshape_op->outputs = expand_op->outputs; in Run()79 std::string axis_array_name = expand_op->inputs[1]; in Run()99 DeleteOpAndArrays(model, expand_op); in Run()
112 TF::ExpandDimsOp expand_op; in matchAndRewrite() local121 expand_op = llvm::cast<TF::ExpandDimsOp>(prev_op); in matchAndRewrite()126 llvm::dyn_cast<TF::ConstOp>(expand_op.dim().getDefiningOp())) { in matchAndRewrite()202 if (expand_op) { in matchAndRewrite()258 if (expand_op) { in matchAndRewrite()265 expand_op.setOperand(0, stb_op.input()); in matchAndRewrite()274 expand_op.getResult().setType(expand_result_type); in matchAndRewrite()