Home
last modified time | relevance | path

Searched defs:size_attr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc227 auto size_attr = size.dyn_cast<DenseIntElementsAttr>(); in CanOptimizeIdentitySliceOp() local
/external/tensorflow/tensorflow/core/transforms/constant_folding/
Dpass.cc915 auto size_attr = array_size->getAttrOfType<SplatElementsAttr>("value"); in matchAndRewrite() local
1560 auto size_attr = size_op->getAttrOfType<ElementsAttr>("value"); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc1604 auto size_attr = DenseIntElementsAttr::get(axis_type, output_shape); in matchAndRewrite() local
Dlegalize_hlo.cc593 auto size_attr = rewriter.create<ConstOp>( in SliceNegativePadding() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3095 DenseIntElementsAttr size_attr; in matchAndRewrite() local