Searched refs:tensor_bytes (Results 1 – 10 of 10) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_striding_sycl.cpp | 46 std::size_t tensor_bytes = tensor.size() * sizeof(DataType); in test_simple_striding() local
49 DataType * d_tensor = static_cast<DataType*>(sycl_device.allocate(tensor_bytes)); in test_simple_striding()
64 sycl_device.memcpyHostToDevice(d_tensor, tensor.data(), tensor_bytes); in test_simple_striding()
128 std::size_t tensor_bytes = tensor.size() * sizeof(DataType); in test_striding_as_lvalue() local
132 DataType * d_tensor = static_cast<DataType*>(sycl_device.allocate(tensor_bytes)); in test_striding_as_lvalue()
150 sycl_device.memcpyHostToDevice(d_tensor, tensor.data(), tensor_bytes); in test_striding_as_lvalue()
|
| D | cxx11_tensor_gpu.cu | 30 std::size_t tensor_bytes = in1.size() * sizeof(float); in test_gpu_nullary() local
34 gpuMalloc((void**)(&d_in1), tensor_bytes); in test_gpu_nullary()
35 gpuMalloc((void**)(&d_in2), tensor_bytes); in test_gpu_nullary()
36 gpuMemcpy(d_in1, in1.data(), tensor_bytes, gpuMemcpyHostToDevice); in test_gpu_nullary()
37 gpuMemcpy(d_in2, in2.data(), tensor_bytes, gpuMemcpyHostToDevice); in test_gpu_nullary()
53 assert(gpuMemcpyAsync(new1.data(), d_in1, tensor_bytes, gpuMemcpyDeviceToHost, in test_gpu_nullary()
55 assert(gpuMemcpyAsync(new2.data(), d_in2, tensor_bytes, gpuMemcpyDeviceToHost, in test_gpu_nullary()
|
| /external/executorch/backends/vulkan/serialization/ |
| D | vulkan_graph_serialize.py | 170 tensor_bytes = bytes(array)
172 raw_bytes += tensor_bytes
173 raw_bytes += b"\x00" * padding_required(len(tensor_bytes))
175 vk_graph.constants.append(VkBytes(current_offset, len(tensor_bytes)))
176 current_offset += aligned_size(len(tensor_bytes))
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | utils_impl.py | 300 tensor_bytes = tensor.tensor_content
301 if tensor_bytes:
305 chunksize = int(len(tensor_bytes) / tensor_size)
308 tensor_bytes[i:i + chunksize]
309 for i in range(0, len(tensor_bytes), chunksize)
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | test_runner.h | 112 bool CheckSizes(size_t tensor_bytes, size_t num_values) { in CheckSizes() argument
113 size_t num_tensor_elements = tensor_bytes / sizeof(T); in CheckSizes()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | test_runner.h | 107 bool CheckSizes(size_t tensor_bytes, size_t num_values) { in CheckSizes() argument
108 size_t num_tensor_elements = tensor_bytes / sizeof(T); in CheckSizes()
|
| /external/executorch/backends/vulkan/test/ |
| D | test_serialization.py | 94 tensor_bytes = bytes(array)
95 self.assertEqual(constant_data_bytes, tensor_bytes)
|
| /external/executorch/backends/qualcomm/runtime/ |
| D | SharedBuffer.cpp | 23 hash_val ^= std::hash<size_t>()(info.tensor_bytes); in operator ()()
37 lhs.pos == rhs.pos && lhs.tensor_bytes == rhs.tensor_bytes && in operator ==()
|
| D | QnnExecuTorch.h | 44 size_t tensor_bytes; member
|
| D | QnnManager.cpp | 132 info.tensor_bytes, in PreRegisterMem()
|