Searched refs:inputOp (Results 1 – 3 of 3) sorted by relevance
99 final Operation inputOp = g.operation(inputName); in create() local100 if (inputOp == null) { in create()103 d.inputSize = (int) inputOp.output(0).shape().size(1); in create()
100 final Operation inputOp = g.operation(d.inputName); in create() local101 if (inputOp == null) { in create()
415 if (auto inputOp = input.getDefiningOp<ShapeOfOp>()) { in getShapeVec() local416 auto type = inputOp.arg().getType().dyn_cast<ShapedType>(); in getShapeVec()421 } else if (auto inputOp = input.getDefiningOp<ConstShapeOp>()) { in getShapeVec() local422 shapeValues = llvm::to_vector<6>(inputOp.shape().getValues<int64_t>()); in getShapeVec()