iree/compiler/Dialect/Stream/Conversion/FlowToStream/ConvertFlowToStream.cpp
// Inserts a sizeof calculation for the given tensor value type and dims.
// This should only be used to produce sizes for values produced by an op; the
// size of operands must be queried from the input resource.
Inserts a sizeof calculation for the given tensor value type and dims.
This should only be used to produce sizes for values produced by an op; the
size of operands must be queried from the input resource.
Inserts a sizeof calculation for the given tensor value type and dims . This should only be used to produce sizes for values produced by an op ; the size of operands must be queried from the input resource .
static Value buildResultSizeOf(Location loc, Value tensorValue,
                               ValueRange dynamicDims,
                               ConversionPatternRewriter &rewriter) {
  return rewriter.createOrFold<IREE::Stream::TensorSizeOfOp>(
      loc, rewriter.getIndexType(), TypeAttr::get(tensorValue.getType()),
      dynamicDims, nullptr);
}
static Value buildResultSizeOf ( Location loc , Value tensorValue , ValueRange dynamicDims , ConversionPatternRewriter & rewriter ) { return rewriter . createOrFold < IREE :: Stream :: TensorSizeOfOp > ( loc , rewriter . getIndexType ( ) , TypeAttr :: get ( tensorValue . getType ( ) ) , dynamicDims , nullptr ) ; }
Inserts a sizeof calculation for the given tensor value type and dims.
Inserts a sizeof calculation for the given tensor value type and dims .
// TODO(benvanik): see if we can stash this on the side to avoid expensive // materialization of a bunch of redundant IR. /*affinity=*/