Home
last modified time | relevance | path

Searched defs:t_in (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/executorch/kernels/test/
H A Dop_squeeze_copy_test.cpp47 Tensor t_in = tf_int.ones({2}); in TEST_F() local
56 Tensor t_in = tf.ones({}); variable
68 Tensor t_in = tf.ones({}); variable
77 Tensor t_in = tf.ones({1}); variable
89 Tensor t_in = tf.ones({2, 1}); variable
101 Tensor t_in = tf.ones({2, 1}); variable
114 Tensor t_in = tf.ones({2, 1}); variable
129 Tensor t_in = tf.ones({2, 1}); variable
142 Tensor t_in = tf.ones({2, 1}); variable
155 Tensor t_in = tf.ones({2, 1}); variable
[all …]
H A Dop_t_copy_test.cpp34 Tensor t_in = tf.make({4}, {1, 2, 3, 4}); variable
47 Tensor t_in = tf.make({4}, {1, 2, 3, 4}); variable
56 Tensor t_in = tf.make({2, 3}, {1, 2, 3, 4, 5, 6}); variable
70 Tensor t_in = tf.make({2, 3}, {1, 2, 3, 4, 5, 6}); variable
79 Tensor t_in = tf.make({2, 3, 1}, {1, 2, 3, 4, 5, 6}); variable
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DCat.cpp27 vTensorPtr t_in = graph.get_tensor(input_ref); in add_cat_default_node() local
42 vTensorPtr t_in = graph.get_tensor(input_ref); in add_cat_default_node() local
54 vTensorPtr t_in = graph.get_tensor(input_ref); in add_cat_default_node() local
65 vTensorPtr t_in = graph.get_tensor(input_ref); in add_cat_default_node() local
76 vTensorPtr t_in = graph.get_tensor(input_ref); in add_cat_default_node() local
H A DIndexSelect.cpp34 vTensorPtr t_in = graph.get_tensor(in); in add_index_select_channel_node() local
81 vTensorPtr t_in = graph.get_tensor(in); in add_index_select_node() local
104 vTensorPtr t_in = graph.get_tensor(in); in get_dim_idx() local
H A DSelect.cpp21 const api::vTensor& t_in, in check_args()
53 vTensorPtr t_in = graph.get_tensor(in); in add_select_int_node() local
H A DSplit.cpp26 vTensorPtr t_in = graph.get_tensor(in); in add_split_with_sizes_default_node() local
124 vTensorPtr t_in = graph.get_tensor(in); in add_split_tensor_node() local
H A DRepeat.cpp68 vTensorPtr t_in = graph.get_tensor(in); in add_repeat_channel_node() local
126 vTensorPtr t_in = graph.get_tensor(in); in add_repeat_node() local
H A DPool.cpp74 vTensorPtr t_in = graph.get_tensor(in); in add_max_pool2d_node() local
151 vTensorPtr t_in = graph.get_tensor(in); in add_avg_pool2d_node() local
H A DCopy.cpp28 vTensorPtr t_in = graph.get_tensor(in); in add_copy_offset_node() local
73 vTensorPtr t_in = graph.get_tensor(in); in add_copy_channel_offset_node() local
H A DConvolution.cpp344 vTensorPtr t_in = graph.get_tensor(in); in add_conv2d_node() local
428 vTensorPtr t_in = graph.get_tensor(in); in add_conv1d_node() local
H A DUnsqueeze.cpp22 vTensorPtr t_in = graph.get_tensor(in); in add_unsqueeze_node() local
H A DSqueeze.cpp23 vTensorPtr t_in = graph.get_tensor(in); in add_squeeze_copy_dims_node() local
H A DEmbedding.cpp35 vTensorPtr t_in = graph.get_tensor(in); in add_embedding_node() local
H A DUpsample.cpp66 vTensorPtr t_in = graph.get_tensor(in); in add_upsample_nearest2d_node() local
H A DView.cpp63 vTensorPtr t_in = graph.get_tensor(in); in add_view_node() local
H A DFlip.cpp51 vTensorPtr t_in = graph.get_tensor(in); in add_flip_node() local
H A DGridPriors.cpp42 vTensorPtr t_in = graph.get_tensor(in); in add_grid_priors_node() local
H A DBatchNorm.cpp73 vTensorPtr t_in = graph.get_tensor(in_ref); in add_native_batch_norm_node() local
H A DPad.cpp65 vTensorPtr t_in = graph.get_tensor(in); in add_constant_pad_nd_node() local
H A DPermute.cpp47 vTensorPtr t_in = graph.get_tensor(in); in add_permute_node() local
H A DSlice.cpp44 vTensorPtr t_in = graph.get_tensor(in); in add_slice_tensor_copy_node() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DCommon.h70 uint32_t get_dim(const Tensor& t_in) { in get_dim()
H A DUtils.cpp99 Tensor nc4hw_to_nchw(const Tensor& t_in, IntArrayRef sizes) { in nc4hw_to_nchw()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dvol2col.cuh48 auto t_in = t_out * stride_t - pad_t; in vol2col_kernel() local
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupMPI.cpp88 const at::Tensor& t_in, in checkSameSizeAndType()

12