Home
last modified time | relevance | path

Searched refs:shape_const (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc721 auto shape_const = in matchAndRewrite() local
724 rewriter.replaceOp(op, {shape_const.getResult()}); in matchAndRewrite()
Dlegalize_tfl.cc1639 auto shape_const = in matchAndRewrite() local
1642 rewriter.replaceOp(op, {shape_const.getResult()}); in matchAndRewrite()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc4088 Output shape_const = ops::Const(scope.WithOpName("shape"), in TEST_F() local
4091 ops::Fill(scope.WithOpName("fill_huge"), shape_const, value); in TEST_F()