Lines Matching refs:getShape

346   ShapeAdaptor inputShape = operands.getShape(0);  in inferReturnTypeComponents()
377 ShapeAdaptor operandShape = operands.getShape(operand); in inferReturnTypeComponents()
406 ShapeAdaptor operandShape = operands.getShape(operand); in inferReturnTypeComponents()
428 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
429 ShapeAdaptor weightShape = operands.getShape(1); in inferReturnTypeComponents()
430 ShapeAdaptor biasShape = operands.getShape(2); in inferReturnTypeComponents()
460 ShapeAdaptor lhsShape = operands.getShape(0); in inferReturnTypeComponents()
461 ShapeAdaptor rhsShape = operands.getShape(1); in inferReturnTypeComponents()
487 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
488 ShapeAdaptor paddingShape = operands.getShape(1); in inferReturnTypeComponents()
558 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
576 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
609 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
647 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents()
648 ShapeAdaptor permsShape = operands.getShape(1); in inferReturnTypeComponents()
717 ShapeAdaptor valuesShape = operands.getShape(0); in inferReturnTypeComponents()
723 ShapeAdaptor indicesShape = operands.getShape(1); in inferReturnTypeComponents()
746 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents()
813 ShapeAdaptor valuesInShape = operands.getShape(0); in inferReturnTypeComponents()
820 ShapeAdaptor indicesShape = operands.getShape(1); in inferReturnTypeComponents()
826 ShapeAdaptor inputShape = operands.getShape(2); in inferReturnTypeComponents()
860 return ReduceInferReturnTypes(operands.getShape(0), \
877 auto shape = operands.getShape(i); in REDUCE_SHAPE_INFER()
887 auto shape = operands.getShape(i); in REDUCE_SHAPE_INFER()
974 ShapeAdaptor inputShape = operands.getShape(0); in NARY_SHAPE_INFER()
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()
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()
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()
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()