Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ops/
H A Dutils.cpp81 int64_t squeeze_dim) { in BuildSqueezedDimensions()
95 int64_t squeeze_dim) { in BuildUnsqueezedDimensions()
/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dops.py869 def squeeze_dim(func, *args, **kwargs): function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dprepare_tf.cc1306 auto squeeze_dim = rewriter.getI64ArrayAttr({-2}); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tf.cc756 for (auto& squeeze_dim : squeeze_dims_attr) { in matchAndRewrite() local
H A Dlegalize_tfl.cc1745 for (auto& squeeze_dim : squeeze_dims_attr) { in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_n_z.cc1652 int64_t squeeze_dim = squeeze_dim_apint.getSExtValue(); in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc6477 int64_t squeeze_dim = squeeze_dim_apint.getSExtValue(); in matchAndRewrite() local
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_lazy_ops.cpp7055 int squeeze_dim = 2; in TEST_F() local