Home
last modified time | relevance | path

Searched defs:expected_dim (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/utils/
H A DParamUtils.h13 int64_t expected_dim) { in _expand_param_if_needed()
30 int64_t expected_dim) { in expand_param_if_needed()
37 int64_t expected_dim) { in expand_param_if_needed()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference_testutil.cc154 StringPiece expected_dim(expected_dims[j]); in InferShapes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/runtime/
H A Darguments.cc154 int64_t expected_dim = std::get<1>(pair.value()); in VerifyMemrefArgument() local
/aosp_15_r20/external/executorch/runtime/core/exec_aten/util/
H A Dtensor_util.h790 size_t expected_dim = static_cast<size_t>(expected_sizes.size()); in tensor_has_expected_size() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dshape_inference.cpp80 int64_t expected_dim) { in expand_param_if_needed()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc2143 int64_t expected_dim = fft_length; in matchAndRewrite() local