Searched refs:tensor_offset (Results 1 – 9 of 9) sorted by relevance
| /external/executorch/backends/qualcomm/runtime/backends/ |
| D | QnnMemManager.cpp | 62 size_t tensor_offset) { in RegisterCustomMem() argument
76 QnnHtpMem_SharedBufferConfig_t htpSharedBuffConfig = {mem_fd, tensor_offset}; in RegisterCustomMem()
105 size_t tensor_offset, in PreRegisterCustomMemHandle() argument
120 QnnHtpMem_SharedBufferConfig_t htpSharedBuffConfig = {mem_fd, tensor_offset}; in PreRegisterCustomMemHandle()
|
| D | QnnMemManager.h | 41 size_t tensor_offset);
49 size_t tensor_offset,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_mul_op.cc | 192 int32_t tensor_offset, int64_t tensor_num_elements, in VectorTensorMultiply() argument
197 (static_cast<int32>(tensor_data[i]) - tensor_offset); in VectorTensorMultiply()
205 const quint8* tensor_data, int32 tensor_offset, int64 tensor_num_elements, in VectorTensorMultiply() argument
208 const uint8x8_t offset_y_8x8 = vmov_n_u8(tensor_offset); in VectorTensorMultiply()
272 (static_cast<int32>(tensor_data[i]) - tensor_offset); in VectorTensorMultiply()
347 int32_t tensor_offset; in Compute() local
354 tensor_offset = offset_y; in Compute()
361 tensor_offset = offset_x; in Compute()
370 tensor_offset, tensor_num_elements, z_data); in Compute()
|
| D | depthwise_conv_op_gpu.h | 234 const int tensor_offset =
285 tile_ptr[tile_offset] = static_cast<S>(ldg(tensor_offset + in_ptr));
321 out_ptr[tensor_offset] = static_cast<T>(sum2);
1202 const int tensor_offset = block_height * in_row_size;
1256 tile_ptr[tile_offset] = static_cast<S>(ldg(tensor_offset + in_ptr));
1268 skip_second ? S() : static_cast<S>(ldg(tensor_offset + out_ptr));
|
| /external/pytorch/torch/multiprocessing/ |
| D | reductions.py | 127 tensor_offset, argument
140 tensor_offset,
159 tensor_offset, argument
208 tensor_offset,
355 tensor_offset = tensor.storage_offset()
365 tensor_offset, # tensor offset in its storage
|
| /external/executorch/backends/qualcomm/runtime/ |
| D | QnnManager.cpp | 112 size_t tensor_offset = (static_cast<char*>(info.custom_mem) - in PreRegisterMem() local
144 tensor_offset, in PreRegisterMem()
249 size_t tensor_offset = static_cast<char*>(custom_mem_base) - in RegisterCustomMem() local
269 tensor_offset) == Error::Ok, in RegisterCustomMem()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-017.md | 29 int32 tensor_offset, int64 tensor_num_elements,
|
| /external/executorch/extension/flat_tensor/ |
| D | flat_tensor.fbs | 38 // 3. tensor_offset: segments[segment_offset].tensor_metadata[j].offset
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 4979 const size_t tensor_offset = static_unpacked_data_.size(); in PrepareOpsToDelegate() local
4980 static_unpacked_data_.resize(tensor_offset + context->tensors[t].bytes); in PrepareOpsToDelegate()
4982 char* unpacked_data = static_unpacked_data_.data() + tensor_offset; in PrepareOpsToDelegate()
5091 static_unpacked_data_map_[t] = tensor_offset; in PrepareOpsToDelegate()
|