| /external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| D | NativeLayerNorm.cpp | 35 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_native_layer_norm_node()
36 vTensorPtr mean = graph->get_tensor(args[0].refs[1]); in resize_native_layer_norm_node()
37 vTensorPtr rstd = graph->get_tensor(args[0].refs[2]); in resize_native_layer_norm_node()
38 vTensorPtr in = graph->get_tensor(args[1].refs[0]); in resize_native_layer_norm_node()
82 vTensorPtr t_out = graph.get_tensor(out_val->at(0)); in add_native_layer_norm_node()
83 vTensorPtr t_mean = graph.get_tensor(out_val->at(1)); in add_native_layer_norm_node()
84 vTensorPtr t_input = graph.get_tensor(in); in add_native_layer_norm_node()
|
| D | IndexSelect.cpp | 34 vTensorPtr t_in = graph.get_tensor(in); in add_index_select_channel_node()
35 vTensorPtr t_idx = graph.get_tensor(idx); in add_index_select_channel_node()
36 vTensorPtr t_out = graph.get_tensor(out); in add_index_select_channel_node()
81 vTensorPtr t_in = graph.get_tensor(in); in add_index_select_node()
82 vTensorPtr t_idx = graph.get_tensor(idx); in add_index_select_node()
83 vTensorPtr t_out = graph.get_tensor(out); in add_index_select_node()
104 vTensorPtr t_in = graph.get_tensor(in); in get_dim_idx()
|
| D | Cat.cpp | 27 vTensorPtr t_in = graph.get_tensor(input_ref); in add_cat_default_node()
32 vTensorPtr t_out = graph.get_tensor(out); in add_cat_default_node()
42 vTensorPtr t_in = graph.get_tensor(input_ref); in add_cat_default_node()
54 vTensorPtr t_in = graph.get_tensor(input_ref); in add_cat_default_node()
65 vTensorPtr t_in = graph.get_tensor(input_ref); in add_cat_default_node()
76 vTensorPtr t_in = graph.get_tensor(input_ref); in add_cat_default_node()
|
| D | Transpose.cpp | 26 vTensorPtr out = graph->get_tensor(extra_args[0]); in resize_transpose_view_node()
27 vTensorPtr in = graph->get_tensor(extra_args[1]); in resize_transpose_view_node()
65 const vTensorPtr in = graph.get_tensor(input_ref); in add_transpose_view_node()
66 graph.get_tensor(out_ref)->virtual_clone(*in); in add_transpose_view_node()
67 graph.get_tensor(out_ref)->virtual_transpose(dim0, dim1); in add_transpose_view_node()
|
| D | Split.cpp | 26 vTensorPtr t_in = graph.get_tensor(in); in add_split_with_sizes_default_node()
40 vTensorPtr t_out = graph.get_tensor(out_ref); in add_split_with_sizes_default_node()
52 vTensorPtr t_out = graph.get_tensor(out_ref); in add_split_with_sizes_default_node()
63 vTensorPtr t_out = graph.get_tensor(out_ref); in add_split_with_sizes_default_node()
74 vTensorPtr t_out = graph.get_tensor(out_ref); in add_split_with_sizes_default_node()
85 vTensorPtr t_out = graph.get_tensor(out_ref); in add_split_with_sizes_default_node()
124 vTensorPtr t_in = graph.get_tensor(in); in add_split_tensor_node()
|
| D | BinaryOp.cpp | 35 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_binary_op_node()
38 vTensorPtr self = graph->get_tensor(args[1].refs[0]); in resize_binary_op_node()
39 vTensorPtr other = graph->get_tensor(args[1].refs[1]); in resize_binary_op_node()
57 vTensorPtr t_in1 = graph.get_tensor(arg1); in add_binary_op_node()
58 vTensorPtr t_in2 = graph.get_tensor(arg2); in add_binary_op_node()
59 vTensorPtr t_out = graph.get_tensor(out); in add_binary_op_node()
|
| D | Pool.cpp | 31 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_pool2d_node()
32 vTensorPtr self = graph->get_tensor(args[1].refs[0]); in resize_pool2d_node()
56 vTensorPtr indices = graph->get_tensor(args[0].refs[1]); in resize_pool2d_node()
74 vTensorPtr t_in = graph.get_tensor(in); in add_max_pool2d_node()
77 vTensorPtr t_out = graph.get_tensor(out_val->at(0)); in add_max_pool2d_node()
151 vTensorPtr t_in = graph.get_tensor(in); in add_avg_pool2d_node()
152 vTensorPtr t_out = graph.get_tensor(out); in add_avg_pool2d_node()
|
| D | GridPriors.cpp | 26 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_grid_priors_node()
27 vTensorPtr in = graph->get_tensor(extra_args[0]); in resize_grid_priors_node()
41 vTensorPtr t_out = graph.get_tensor(out); in add_grid_priors_node()
42 vTensorPtr t_in = graph.get_tensor(in); in add_grid_priors_node()
|
| D | Convolution.cpp | 26 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_conv2d_node()
27 vTensorPtr self = graph->get_tensor(args[1].refs[0]); in resize_conv2d_node()
61 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_conv1d_node()
62 vTensorPtr self = graph->get_tensor(args[1].refs[0]); in resize_conv1d_node()
98 vTensorPtr t = graph.get_tensor(v); in prepack_biases()
203 vTensorPtr t = graph.get_tensor(v); in prepack_weights()
344 vTensorPtr t_in = graph.get_tensor(in); in add_conv2d_node()
345 vTensorPtr t_out = graph.get_tensor(out); in add_conv2d_node()
428 vTensorPtr t_in = graph.get_tensor(in); in add_conv1d_node()
429 vTensorPtr t_weight = graph.get_tensor(arg_weight); in add_conv1d_node()
[all …]
|
| D | Flip.cpp | 28 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_flip_node()
29 vTensorPtr in = graph->get_tensor(args[1].refs[0]); in resize_flip_node()
51 vTensorPtr t_in = graph.get_tensor(in); in add_flip_node()
52 vTensorPtr t_out = graph.get_tensor(out); in add_flip_node()
|
| D | Pad.cpp | 44 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_constant_pad_node()
45 vTensorPtr self = graph->get_tensor(args[1].refs[0]); in resize_constant_pad_node()
65 vTensorPtr t_in = graph.get_tensor(in); in add_constant_pad_nd_node()
66 vTensorPtr t_out = graph.get_tensor(out); in add_constant_pad_nd_node()
|
| D | View.cpp | 46 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_view_node()
47 vTensorPtr in = graph->get_tensor(args[1].refs[0]); in resize_view_node()
63 vTensorPtr t_in = graph.get_tensor(in); in add_view_node()
64 vTensorPtr t_out = graph.get_tensor(out); in add_view_node()
|
| D | BatchNorm.cpp | 49 std::vector<int64_t> in_sizes = graph.get_tensor(in_ref)->sizes(); in add_native_batch_norm_node()
50 std::vector<int64_t> out_sizes = graph.get_tensor(in_ref)->sizes(); in add_native_batch_norm_node()
73 vTensorPtr t_in = graph.get_tensor(in_ref); in add_native_batch_norm_node()
76 vTensorPtr t_out = graph.get_tensor(out_ref); in add_native_batch_norm_node()
|
| D | Upsample.cpp | 23 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_upsample_nearest2d_node()
24 vTensorPtr self = graph->get_tensor(args[1].refs[0]); in resize_upsample_nearest2d_node()
66 vTensorPtr t_in = graph.get_tensor(in); in add_upsample_nearest2d_node()
91 vTensorPtr t_out = graph.get_tensor(out); in add_upsample_nearest2d_node()
|
| D | Embedding.cpp | 34 vTensorPtr t_weight = graph.get_tensor(weight); in add_embedding_node()
35 vTensorPtr t_in = graph.get_tensor(in); in add_embedding_node()
36 vTensorPtr t_out = graph.get_tensor(out); in add_embedding_node()
|
| D | Full.cpp | 22 vTensorPtr out = graph->get_tensor(args[0].refs[0]); in resize_full_node()
25 out_sizes = graph->get_tensor(extra_args[0])->sizes(); in resize_full_node()
40 vTensorPtr t_out = graph.get_tensor(out); in add_full_node()
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuDepthwiseConv2d.cpp | 153 auto dst = tensors.get_tensor(TensorType::ACL_DST_0); in run()
154 auto workspace = tensors.get_tensor(TensorType::ACL_INT_3); in run()
155 auto packed_weights = tensors.get_tensor(TensorType::ACL_INT_4); in run()
162 auto src_perm = tensors.get_tensor(TensorType::ACL_INT_0); in run()
171 auto src_perm = tensors.get_tensor(TensorType::ACL_INT_0); in run()
172 auto weights_perm = tensors.get_tensor(TensorType::ACL_INT_1); in run()
173 auto dst_perm = tensors.get_tensor(TensorType::ACL_INT_2); in run()
186 auto src = tensors.get_tensor(TensorType::ACL_SRC_0); in run()
187 auto weights = tensors.get_tensor(TensorType::ACL_SRC_1); in run()
188 auto dst = tensors.get_tensor(TensorType::ACL_DST); in run()
[all …]
|
| /external/pytorch/test/cpp_extensions/ |
| D | maia_extension.cpp | 8 Tensor get_tensor(caffe2::TypeMeta dtype, IntArrayRef size) { in get_tensor() function
26 return get_tensor(scalarTypeToTypeMeta(dtype_or_default(dtype)), size); in empty_override()
40 return get_tensor(input.dtype(), {input.size(0), weight.size(0), input.size(2), input.size(3)}); in fake_convolution()
50 get_tensor(input.dtype(), input.sizes()), in fake_convolution_backward()
51 get_tensor(weight.dtype(), weight.sizes()), in fake_convolution_backward()
52 get_tensor(input.dtype(), {})); in fake_convolution_backward()
|
| /external/executorch/backends/qualcomm/builders/ |
| D | op_index_put.py | 24 input_tensor = self.get_tensor(input_node, node)
34 self.get_tensor(idx, idx) for idx in indicies_node if idx is not None
55 value_tensor = self.get_tensor(value_node, node)
64 output_tensor = self.get_tensor(node, node)
|
| D | op_to.py | 47 input_tensor = self.get_tensor(input_node, node)
51 output_tensor = self.get_tensor(node, node)
83 input_tensor = self.get_tensor(input_node, node)
93 output_tensor = self.get_tensor(node, node)
|
| D | op_index.py | 31 input_tensor = self.get_tensor(input_node, node)
45 indices_tensor = self.get_tensor(indices_node, node).to(torch.int32)
58 output_tensor = self.get_tensor(node, node)
|
| D | op_topk.py | 33 input_tensor = self.get_tensor(input_node, node)
59 output_val_tensor = self.get_tensor(node, node, 0)
60 output_idx_tensor = self.get_tensor(node, node, 1).to(torch.int32)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | py_checkpoint_reader.py | 63 def get_tensor(self, tensor_str): function
74 CheckpointReader.get_tensor = get_tensor
|
| /external/ComputeLibrary/src/common/ |
| D | TensorPack.cpp | 65 arm_compute::ITensor *TensorPack::get_tensor(int32_t slot_id) in get_tensor() function in arm_compute::TensorPack
67 return _pack.get_tensor(slot_id); in get_tensor()
|
| /external/executorch/backends/vulkan/runtime/graph/ops/ |
| D | PrepackNode.cpp | 21 vTensorPtr t_packed = graph.get_tensor(packed); in get_noop_shader()
49 vTensorPtr packed = graph->get_tensor(packed_); in create_staging_buffer()
71 vTensorPtr packed = graph->get_tensor(packed_); in encode()
|