Home
last modified time | relevance | path

Searched defs:updates_type (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcollection_ops_util.cc292 auto updates_type = updates.getType().cast<RankedTensorType>(); in ScatterAccumulateElements() local
Dlegalize_hlo.cc2801 ShapedType &updates_type, ConversionPatternRewriter &rewriter) { in CanonicalizeScatterUpdates()
3141 ShapedType updates_type = updates[0].getType().cast<ShapedType>(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1416 auto updates_type = updates.getType().cast<ShapedType>(); in verify() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc6012 auto updates_type = updates.getType().cast<ShapedType>(); in matchAndRewrite() local