Home
last modified time | relevance | path

Searched defs:expected_output_size (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dbroadcast_util.h135 Tensor::SizesType expected_output_size[kTensorDimensionLimit]; in resize_to_broadcast_target_size() local
177 Tensor::SizesType expected_output_size[kTensorDimensionLimit]; in resize_to_broadcast_target_size() local
H A Dkernel_ops_util.cpp559 Tensor::SizesType expected_output_size[kTensorDimensionLimit]; in resize_constant_pad_output() local
603 Tensor::SizesType expected_output_size[kTensorDimensionLimit]; in resize_embedding_output() local
H A Dactivation_ops_util.cpp80 exec_aten::SizesType expected_output_size[kTensorDimensionLimit]; in resize_glu_out() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dspectrogram_test.cc124 for (int expected_output_size : expected_output_sizes) { in TEST() local
147 for (int expected_output_size : expected_output_sizes) { in TEST() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_view_copy.cpp29 Tensor::SizesType expected_output_size[16]; in view_copy_out() local
H A Dop_unsqueeze_copy.cpp29 Tensor::SizesType expected_output_size[kTensorDimensionLimit]; in unsqueeze_copy_out() local
H A Dop_repeat.cpp57 Tensor::SizesType expected_output_size[kTensorDimensionLimit]; in repeat_out() local
/aosp_15_r20/external/executorch/kernels/prim_ops/
H A Det_copy_index.cpp69 SizesType expected_output_size[kTensorDimensionLimit]; in et_copy_index() local
H A Det_view.cpp77 SizesType expected_output_size[kTensorDimensionLimit]; in et_view() local
/aosp_15_r20/external/executorch/kernels/test/custom_kernel_example/
H A Dop_relu.cpp40 Tensor::SizesType expected_output_size[16]; in relu() local
/aosp_15_r20/external/executorch/backends/xnnpack/runtime/
H A DXNNExecutor.cpp207 SizesType expected_output_size[kTensorDimensionLimit]; in resize_outputs() local
/aosp_15_r20/external/executorch/kernels/optimized/cpu/
H A Dop_bmm.cpp114 exec_aten::SizesType expected_output_size[kTensorDimensionLimit]; in resize_out_tensor() local
/aosp_15_r20/system/logging/liblog/tests/
H A Dlogprint_test.cpp49 auto expected_output_size = expected_output.size(); in TEST() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/test/
H A Dbroadcast_test.cpp115 expected_output_size[torch::executor::kTensorDimensionLimit] = {}; in TEST() local
/aosp_15_r20/external/mbedtls/library/
H A Dpsa_crypto_cipher.c375 size_t expected_output_size; in mbedtls_psa_cipher_update() local
/aosp_15_r20/external/executorch/kernels/quantized/cpu/
H A Dop_embedding.cpp186 exec_aten::SizesType expected_output_size[kTensorDimensionLimit]; in resize_out_tensor() local
H A Dembeddingxb.cpp228 exec_aten::SizesType expected_output_size[kTensorDimensionLimit]; in resize_out_tensor() local
/aosp_15_r20/external/openthread/third_party/mbedtls/repo/library/
H A Dpsa_crypto_cipher.c505 size_t expected_output_size; in mbedtls_psa_cipher_update() local
/aosp_15_r20/external/executorch/kernels/prim_ops/test/
H A Dprim_ops_test.cpp136 SizesType expected_output_size[2] = {0, 0}; in TEST_F() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDepthwiseConv3d.cu334 auto expected_output_size = conv_output_size(input.sizes(), weight.sizes(), in conv_depthwise_shape_check() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding_test.cc3912 int expected_output_size = SetFetch(&item, fetch_const, fetch_cast, in ConstantFoldingCastConst() local
3932 int expected_output_size = 0; in SetFetch() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc1776 int64_t expected_output_size; in inferConvReturnTypeComponents() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc4972 int64_t expected_output_size = 0; in matchAndRewrite() local