Lines Matching refs:operands
344 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
346 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
369 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
376 for (auto operand : operands) { in inferReturnTypeComponents()
377 ShapeAdaptor operandShape = operands.getShape(operand); in inferReturnTypeComponents()
405 for (auto operand : operands) { in inferReturnTypeComponents()
406 ShapeAdaptor operandShape = operands.getShape(operand); in inferReturnTypeComponents()
426 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
428 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
429 ShapeAdaptor weightShape = operands.getShape(1); in inferReturnTypeComponents()
430 ShapeAdaptor biasShape = operands.getShape(2); in inferReturnTypeComponents()
458 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
460 ShapeAdaptor lhsShape = operands.getShape(0); in inferReturnTypeComponents()
461 ShapeAdaptor rhsShape = operands.getShape(1); in inferReturnTypeComponents()
485 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
487 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
488 ShapeAdaptor paddingShape = operands.getShape(1); in inferReturnTypeComponents()
513 if (!matchPattern(operands[1], m_Constant(&paddings))) { in inferReturnTypeComponents()
541 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
543 ArrayAttr sizes = SliceOpAdaptor(operands, attributes).getSize(); in inferReturnTypeComponents()
556 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
558 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
572 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
574 TileOpAdaptor adaptor(operands, attributes); in inferReturnTypeComponents()
576 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
606 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
608 ReshapeOpAdaptor adaptor(operands, attributes); in inferReturnTypeComponents()
609 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
645 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
647 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
648 ShapeAdaptor permsShape = operands.getShape(1); in inferReturnTypeComponents()
699 if (ShapeAdaptor permShape = operands.getValueAsShape(1)) { in inferReturnTypeComponents()
712 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
717 ShapeAdaptor valuesShape = operands.getShape(0); in inferReturnTypeComponents()
723 ShapeAdaptor indicesShape = operands.getShape(1); in inferReturnTypeComponents()
737 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
739 ResizeOpAdaptor adaptor(operands, attributes); in inferReturnTypeComponents()
746 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents()
808 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
813 ShapeAdaptor valuesInShape = operands.getShape(0); in inferReturnTypeComponents()
820 ShapeAdaptor indicesShape = operands.getShape(1); in inferReturnTypeComponents()
826 ShapeAdaptor inputShape = operands.getShape(2); in inferReturnTypeComponents()
857 ValueShapeRange operands, DictionaryAttr attributes, \
860 return ReduceInferReturnTypes(operands.getShape(0), \
873 static LogicalResult resolveBroadcastShape(const ValueShapeRange &operands, in REDUCE_SHAPE_INFER()
876 for (int i = 0, e = operands.size(); i != e; ++i) { in REDUCE_SHAPE_INFER()
877 auto shape = operands.getShape(i); in REDUCE_SHAPE_INFER()
886 for (int i = 0, e = operands.size(); i != e; ++i) { in REDUCE_SHAPE_INFER()
887 auto shape = operands.getShape(i); in REDUCE_SHAPE_INFER()
910 const ValueShapeRange &operands, in NAryInferReturnTypes() argument
913 if (resolveBroadcastShape(operands, outShape).failed()) { in NAryInferReturnTypes()
924 ValueShapeRange operands, DictionaryAttr attributes, \
927 return NAryInferReturnTypes(operands, inferredReturnShapes); \
972 const ValueShapeRange &operands, DictionaryAttr attributes, in NARY_SHAPE_INFER()
974 ShapeAdaptor inputShape = operands.getShape(0); in NARY_SHAPE_INFER()
1015 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
1018 Conv2DOp::Adaptor adaptor(operands.getValues(), attributes); in inferReturnTypeComponents()
1027 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents()
1035 ShapeAdaptor weightShape = operands.getShape(adaptor.getWeight()); in inferReturnTypeComponents()
1043 ShapeAdaptor biasShape = operands.getShape(adaptor.getBias()); in inferReturnTypeComponents()
1082 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
1085 Conv2DOp::Adaptor adaptor(operands.getValues(), attributes); in inferReturnTypeComponents()
1096 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents()
1105 ShapeAdaptor weightShape = operands.getShape(adaptor.getWeight()); in inferReturnTypeComponents()
1114 ShapeAdaptor biasShape = operands.getShape(adaptor.getBias()); in inferReturnTypeComponents()
1160 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
1162 return poolingInferReturnTypes(operands, attributes, inferredReturnShapes); in inferReturnTypeComponents()
1167 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
1169 return poolingInferReturnTypes(operands, attributes, inferredReturnShapes); in inferReturnTypeComponents()
1174 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
1177 DepthwiseConv2DOp::Adaptor adaptor(operands.getValues(), attributes); in inferReturnTypeComponents()
1188 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents()
1197 ShapeAdaptor weightShape = operands.getShape(adaptor.getWeight()); in inferReturnTypeComponents()
1215 ShapeAdaptor biasShape = operands.getShape(adaptor.getBias()); in inferReturnTypeComponents()
1254 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
1256 TransposeConv2DOp::Adaptor adaptor(operands.getValues(), attributes); in inferReturnTypeComponents()
1266 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents()
1276 ShapeAdaptor weightShape = operands.getShape(adaptor.getFilter()); in inferReturnTypeComponents()
1286 ShapeAdaptor biasShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents()
1319 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument
1363 ValueShapeRange operands, DictionaryAttr attributes, RegionRange regions, in inferReturnTypeComponents() argument