Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc1306 auto squeeze_dim = rewriter.getI64ArrayAttr({-2}); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc756 for (auto& squeeze_dim : squeeze_dims_attr) { in matchAndRewrite() local
Dlegalize_tfl.cc1745 for (auto& squeeze_dim : squeeze_dims_attr) { in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1652 int64_t squeeze_dim = squeeze_dim_apint.getSExtValue(); in verify() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc6477 int64_t squeeze_dim = squeeze_dim_apint.getSExtValue(); in matchAndRewrite() local