Home
last modified time | relevance | path

Searched defs:t_out (Results 1 – 25 of 39) sorted by relevance

12

/aosp_15_r20/external/executorch/kernels/test/
H A Dop_squeeze_copy_test.cpp48 Tensor t_out = tf_d.ones({2}); in TEST_F() local
57 Tensor t_out = tf.zeros({}); variable
69 Tensor t_out = tf.ones({}); variable
78 Tensor t_out = tf.make({}, {99}); variable
90 Tensor t_out = tf.make({2, 1}, {4, 3}); variable
102 Tensor t_out = tf.make({2}, {4, 3}); variable
115 Tensor t_out = tf.zeros( variable
130 Tensor t_out = tf.zeros( variable
143 Tensor t_out = tf.zeros( variable
156 Tensor t_out = tf.zeros( variable
[all …]
H A Dop_t_copy_test.cpp35 Tensor t_out = tf.make({4}, {0, 0, 0, 0}); variable
48 Tensor t_out = tf.make({2}, {0, 0}); variable
57 Tensor t_out = tf.make({3, 2}, {0, 0, 0, 0, 0, 0}); variable
71 Tensor t_out = tf.make({2, 2}, {0, 0, 0, 0}); variable
80 Tensor t_out = tf.make({3, 2, 1}, {0, 0, 0, 0, 0, 0}); variable
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DSplit.cpp40 vTensorPtr t_out = graph.get_tensor(out_ref); in add_split_with_sizes_default_node() local
52 vTensorPtr t_out = graph.get_tensor(out_ref); in add_split_with_sizes_default_node() local
63 vTensorPtr t_out = graph.get_tensor(out_ref); in add_split_with_sizes_default_node() local
74 vTensorPtr t_out = graph.get_tensor(out_ref); in add_split_with_sizes_default_node() local
85 vTensorPtr t_out = graph.get_tensor(out_ref); in add_split_with_sizes_default_node() local
H A DSelect.cpp24 const api::vTensor& t_out) { in check_args()
54 vTensorPtr t_out = graph.get_tensor(out); in add_select_int_node() local
H A DIndexSelect.cpp36 vTensorPtr t_out = graph.get_tensor(out); in add_index_select_channel_node() local
83 vTensorPtr t_out = graph.get_tensor(out); in add_index_select_node() local
H A DConvolution.cpp125 const api::vTensor& t_out, in get_conv2d_shader()
345 vTensorPtr t_out = graph.get_tensor(out); in add_conv2d_node() local
431 vTensorPtr t_out = graph.get_tensor(out); in add_conv1d_node() local
H A DRepeat.cpp69 vTensorPtr t_out = graph.get_tensor(out); in add_repeat_channel_node() local
127 vTensorPtr t_out = graph.get_tensor(out); in add_repeat_node() local
H A DPool.cpp77 vTensorPtr t_out = graph.get_tensor(out_val->at(0)); in add_max_pool2d_node() local
152 vTensorPtr t_out = graph.get_tensor(out); in add_avg_pool2d_node() local
H A DCopy.cpp29 vTensorPtr t_out = graph.get_tensor(out); in add_copy_offset_node() local
74 vTensorPtr t_out = graph.get_tensor(out); in add_copy_channel_offset_node() local
H A DUnsqueeze.cpp23 vTensorPtr t_out = graph.get_tensor(out); in add_unsqueeze_node() local
H A DSqueeze.cpp24 vTensorPtr t_out = graph.get_tensor(out); in add_squeeze_copy_dims_node() local
H A DEmbedding.cpp36 vTensorPtr t_out = graph.get_tensor(out); in add_embedding_node() local
H A DFull.cpp40 vTensorPtr t_out = graph.get_tensor(out); in add_full_node() local
H A DArange.cpp85 vTensorPtr t_out = graph.get_tensor(out); in add_arange_node() local
H A DUpsample.cpp91 vTensorPtr t_out = graph.get_tensor(out); in add_upsample_nearest2d_node() local
H A DView.cpp64 vTensorPtr t_out = graph.get_tensor(out); in add_view_node() local
H A DCat.cpp32 vTensorPtr t_out = graph.get_tensor(out); in add_cat_default_node() local
H A DFlip.cpp52 vTensorPtr t_out = graph.get_tensor(out); in add_flip_node() local
H A DGridPriors.cpp41 vTensorPtr t_out = graph.get_tensor(out); in add_grid_priors_node() local
H A DBinaryOp.cpp59 vTensorPtr t_out = graph.get_tensor(out); in add_binary_op_node() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Drecord_input_op.cc58 auto t_out = out.flat<tstring>(); in Compute() local
H A Din_topk_op.cc70 Tensor* t_out = nullptr; in Compute() local
/aosp_15_r20/external/libwebsockets/lib/roles/http/compression/brotli/
H A Dbrotli.c51 size_t a_in, a_out, t_out; in lcs_process_brotli() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/
H A DKernelFunction_test.cpp330 at::Tensor& t_out = func.call<at::Tensor&, at::Tensor&, at::Scalar>(dummy, CPU_TEST_SET, t, 1.0f); in expectInPlaceUnboxedCallingWorks() local
341 at::Tensor& t_out = func.call<at::Tensor&, at::Scalar, at::Tensor&>(dummy, CPU_TEST_SET, 1.0f, t); in expectOutOfPlaceUnboxedCallingWorks() local
/aosp_15_r20/external/pytorch/android/test_app/app/src/main/cpp/
H A Dpytorch_testapp_jni.cpp46 c10::IValue t_out = module.forward({t}); in loadAndForwardModel() local

12