Home
last modified time | relevance | path

Searched refs:add_tensor (Results 1 – 25 of 133) sorted by relevance

123456

/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dfuse_add_to_conv_test.cc55 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
56 add_tensor.shape = Linear(16); in TEST()
57 add_tensor.data.resize(16); in TEST()
59 add_attr.param = add_tensor; in TEST()
98 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
99 add_tensor.shape = Linear(2); in TEST()
100 add_tensor.data = {0.3f, 0.7f}; in TEST()
102 add_attr.param = add_tensor; in TEST()
119 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
120 add_tensor.shape = Linear(4); in TEST()
[all …]
Dadd_quant_adjustments_test.cc57 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
58 add_tensor.shape = Linear(8); in TEST()
59 add_tensor.data.resize(8); in TEST()
61 add_attr.param = add_tensor; in TEST()
101 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
102 add_tensor.shape = Linear(8); in TEST()
103 add_tensor.data.resize(8); in TEST()
105 add_attr.param = add_tensor; in TEST()
/external/ComputeLibrary/src/cpu/operators/
DCpuDepthwiseConv2d.cpp163 pack.add_tensor(TensorType::ACL_SRC, src); in run()
164 pack.add_tensor(TensorType::ACL_DST, src_perm); in run()
176 pack.add_tensor(TensorType::ACL_SRC_0, src_perm); in run()
177 pack.add_tensor(TensorType::ACL_SRC_1, weights_perm); in run()
178 pack.add_tensor(TensorType::ACL_SRC_2, bias); in run()
179 pack.add_tensor(TensorType::ACL_INT_0, workspace); in run()
180 pack.add_tensor(TensorType::ACL_INT_1, packed_weights); in run()
181 pack.add_tensor(TensorType::ACL_DST, dst_perm); in run()
191 pack.add_tensor(TensorType::ACL_SRC_0, src); in run()
192 pack.add_tensor(TensorType::ACL_SRC_1, weights); in run()
[all …]
DCpuDirectConv2d.cpp128 pack.add_tensor(TensorType::ACL_SRC_DST, src); in run()
135 pack.add_tensor(TensorType::ACL_SRC_0, dst); in run()
136 pack.add_tensor(TensorType::ACL_SRC_1, bias); in run()
137 pack.add_tensor(TensorType::ACL_DST, dst); in run()
144 pack.add_tensor(TensorType::ACL_SRC, dst); in run()
145 pack.add_tensor(TensorType::ACL_DST, dst); in run()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEElementwiseOperations.cpp69 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
70 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
71 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
110 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
111 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
112 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
151 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
152 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
153 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
192 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
[all …]
DNELogical.cpp57 _impl->pack.add_tensor(TensorType::ACL_SRC_0, input1); in configure()
58 _impl->pack.add_tensor(TensorType::ACL_SRC_1, input2); in configure()
59 _impl->pack.add_tensor(TensorType::ACL_DST, output); in configure()
90 _impl->pack.add_tensor(TensorType::ACL_SRC_0, input1); in configure()
91 _impl->pack.add_tensor(TensorType::ACL_SRC_1, input2); in configure()
92 _impl->pack.add_tensor(TensorType::ACL_DST, output); in configure()
123 _impl->pack.add_tensor(TensorType::ACL_SRC_0, input); in configure()
124 _impl->pack.add_tensor(TensorType::ACL_DST, output); in configure()
DNEDepthwiseConvolutionLayer.cpp168 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src); in run()
169 pack.add_tensor(TensorType::ACL_SRC_1, _impl->weights); in run()
170 pack.add_tensor(TensorType::ACL_SRC_2, _impl->biases); in run()
171 pack.add_tensor(TensorType::ACL_INT_0, &_impl->permuted_input); in run()
172 pack.add_tensor(TensorType::ACL_INT_1, &_impl->permuted_weights); in run()
173 pack.add_tensor(TensorType::ACL_INT_2, &_impl->permuted_output); in run()
174 pack.add_tensor(TensorType::ACL_INT_3, &_impl->workspace); in run()
175 pack.add_tensor(TensorType::ACL_INT_4, &_impl->packed_weights); in run()
176 pack.add_tensor(TensorType::ACL_DST_0, _impl->dst); in run()
284 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src); in run()
[all …]
DNEPixelWiseMultiplication.cpp66 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
67 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
68 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
103 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
104 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
105 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
DNEDirectConvolutionLayer.cpp68 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src); in run()
69 pack.add_tensor(TensorType::ACL_SRC_1, _impl->weights); in run()
70 pack.add_tensor(TensorType::ACL_SRC_2, _impl->bias); in run()
71 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
DNEScale.cpp126 pack.add_tensor(TensorType::ACL_SRC, _impl->src); in run()
127 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
128 pack.add_tensor(TensorType::ACL_INT_0, &_impl->dx); in run()
129 pack.add_tensor(TensorType::ACL_INT_1, &_impl->dy); in run()
130 pack.add_tensor(TensorType::ACL_INT_2, &_impl->offsets); in run()
DNEPReluLayer.cpp61 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
62 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
63 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
DNEArithmeticSubtraction.cpp66 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
67 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
68 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
DNEArithmeticAddition.cpp66 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
67 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
68 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLElementwiseOperations.cpp76 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
77 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
78 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
122 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
123 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
124 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
167 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
168 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
169 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
212 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
[all …]
DCLElementwiseUnaryLayer.cpp70 pack.add_tensor(TensorType::ACL_SRC, _impl->src); in run()
71 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
112 pack.add_tensor(TensorType::ACL_SRC, _impl->src); in run()
113 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
153 pack.add_tensor(TensorType::ACL_SRC, _impl->src); in run()
154 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
194 pack.add_tensor(TensorType::ACL_SRC, _impl->src); in run()
195 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
235 pack.add_tensor(TensorType::ACL_SRC, _impl->src); in run()
236 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
[all …]
DCLPixelWiseMultiplication.cpp76 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
77 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
78 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
121 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
122 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
123 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
DCLConv3D.cpp76 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src); in run()
77 pack.add_tensor(TensorType::ACL_SRC_1, _impl->weights); in run()
78 pack.add_tensor(TensorType::ACL_SRC_2, _impl->biases); in run()
79 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
DCLIndirectConvolutionLayer.cpp80 pack.add_tensor(TensorType::ACL_SRC, _impl->src); in run()
81 pack.add_tensor(TensorType::ACL_SRC_1, _impl->weights); in run()
82 pack.add_tensor(TensorType::ACL_SRC_2, _impl->biases); in run()
83 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
DCLDirectConvolutionLayer.cpp84 pack.add_tensor(TensorType::ACL_SRC, _impl->src); in run()
85 pack.add_tensor(TensorType::ACL_SRC_1, _impl->weights); in run()
86 pack.add_tensor(TensorType::ACL_SRC_2, _impl->biases); in run()
87 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
DCLPReluLayer.cpp72 pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0); in run()
73 pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1); in run()
74 pack.add_tensor(TensorType::ACL_DST, _impl->dst); in run()
/external/pytorch/torch/testing/_internal/
Dtorchbind_impls.py38 return foo.add_tensor(z)
53 a = foo.add_tensor(x)
54 b = foo.add_tensor(a)
55 c = foo.add_tensor(b)
59 a = foo.add_tensor(x)
60 b = foo.add_tensor(a)
73 lambda foo, x: foo.add_tensor(x)
88 def add_tensor(self, z): member in register_fake_classes.FakeFoo
/external/ComputeLibrary/src/gpu/cl/operators/
DClSoftmax.cpp149 pack.add_tensor(TensorType::ACL_DST, permuted_src.get()); in run()
158 norm_pack.add_tensor(TensorType::ACL_DST, permuted_dst.get()); in run()
163 norm_pack.add_tensor(TensorType::ACL_DST, dst); in run()
165 sum_pack.add_tensor(TensorType::ACL_DST, tmp.get()); in run()
166 sum_pack.add_tensor(TensorType::ACL_INT_0, max.get()); in run()
167 sum_pack.add_tensor(TensorType::ACL_INT_1, sum.get()); in run()
170 norm_pack.add_tensor(TensorType::ACL_INT_0, sum.get()); in run()
179 pack.add_tensor(TensorType::ACL_DST, dst); in run()
/external/executorch/backends/vulkan/runtime/graph/
DComputeGraph.cpp62 vref(graph_p->add_tensor( in VALUE_PTR_CLASS_IMPL()
76 vref(graph_p->add_tensor(sizes, dtype, storage_type, sobj_idx)) {} in TmpTensor()
85 vref(graph_p->add_tensor(sizes, dtype, memory_layout, sobj_idx)) {} in TmpTensor()
93 vref(graph_p->add_tensor(sizes, dtype, sobj_idx)) {} in TmpTensor()
237 ValueRef ComputeGraph::add_tensor( in add_tensor() function in vkcompute::ComputeGraph
256 ValueRef ComputeGraph::add_tensor( in add_tensor() function in vkcompute::ComputeGraph
261 return add_tensor( in add_tensor()
269 ValueRef ComputeGraph::add_tensor( in add_tensor() function in vkcompute::ComputeGraph
274 return add_tensor( in add_tensor()
282 return add_tensor(sizes_of(idx), dtype_of(idx), storage_type, memory_layout); in add_tensor_like()
[all …]
/external/ComputeLibrary/src/core/
DITensorPack.cpp39 void ITensorPack::add_tensor(int id, ITensor *tensor) in add_tensor() function in arm_compute::ITensorPack
44 void ITensorPack::add_tensor(int id, const ITensor *tensor) in add_tensor() function in arm_compute::ITensorPack
51 add_tensor(id, tensor); in add_const_tensor()
/external/pytorch/test/inductor/
Dtest_binary_folding.py37 def test_conv_fusion(use_bias, module, op, scalar, add_tensor, expect_success): argument
53 add_tensor
54 if add_tensor is not None
121 add_tensor=None,
132 add_tensor=torch.rand(
146 add_tensor=torch.rand(1, 1).to(self.device),
156 add_tensor=torch.tensor([2]).to(torch.float64).to(self.device),

123456