Home
last modified time | relevance | path

Searched refs:tensor_bytes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/accuracy/
Drun_tflite_model_op.cc114 auto tensor_bytes = input_tensors[i].tensor_data(); in Compute() local
115 OP_REQUIRES(context, tflite_tensor->bytes == tensor_bytes.size(), in Compute()
118 " got: ", tensor_bytes.size())); in Compute()
119 std::memcpy(tflite_tensor->data.raw, tensor_bytes.data(), in Compute()
120 tensor_bytes.size()); in Compute()
130 auto tensor_bytes = output->tensor_data(); in Compute() local
131 OP_REQUIRES(context, tensor_bytes.size() == tfl_tensor->bytes, in Compute()
133 std::memcpy(const_cast<char*>(tensor_bytes.data()), tfl_tensor->data.raw, in Compute()
/external/tensorflow/tensorflow/lite/testing/
Dtest_runner.h109 bool CheckSizes(size_t tensor_bytes, size_t num_values) { in CheckSizes() argument
110 size_t num_tensor_elements = tensor_bytes / sizeof(T); in CheckSizes()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_runner.h112 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/contrib/verbs/
Drdma.cc884 size_t tensor_bytes, bool is_gpu_to_cpu) { in CountCopies() argument
894 numGPUToCPUCopiedBytes += tensor_bytes; in CountCopies()
897 numCPUToGPUCopiedBytes += tensor_bytes; in CountCopies()
1168 size_t tensor_bytes = (can_memcpy) ? in.TotalBytes() : proto.ByteSize(); in SendMetaData() local
1179 rm.tensor_bytes_ = tensor_bytes; in SendMetaData()
1199 size_t tensor_bytes = (can_memcpy) ? in.TotalBytes() : proto.ByteSize(); in SendContent() local
1208 tensor_bytes); in SendContent()
1212 << " (Size: " << tensor_bytes << ") " << in.DebugString(); in SendContent()
1213 src_addr_ = malloc(tensor_bytes); in SendContent()
1214 mr_ = ibv_reg_mr(channel_->adapter_->pd_, src_addr_, tensor_bytes, in SendContent()
[all …]
Dpatch_notes_verbs_with_0_copies.md48 … information with the tensor, and thus it is now always equal to the "tensor_bytes" field. Instead…
52 … step_id | request_index | remote_addr | rkey | is_dead | data_type | tensor_shape | tensor_bytes |
62 …* is_dead/data_type/tensor_shape/tensor_bytes - The current meta-data as stored in the receiver lo…
66 * is_dead/data_type/tensor_shape/tensor_bytes - The up-to-date meta-data.
73 …* is_dead/data_type/tensor_shape/tensor_bytes - The new meta-data. Will be removed in the next pha…
DREADME.md140 … | remote_addr/checksum | rkey | is_dead | data_type | tensor_shape | tensor_bytes | error_status …
150 …* is_dead/data_type/tensor_shape/tensor_bytes - The current meta-data as stored in the receiver lo…
154 * is_dead/data_type/tensor_shape/tensor_bytes - The up-to-date meta-data.
/external/eigen/unsupported/test/
Dcxx11_tensor_cuda.cu29 std::size_t tensor_bytes = in1.size() * sizeof(float); in test_cuda_nullary() local
33 cudaMalloc((void**)(&d_in1), tensor_bytes); in test_cuda_nullary()
34 cudaMalloc((void**)(&d_in2), tensor_bytes); in test_cuda_nullary()
35 cudaMemcpy(d_in1, in1.data(), tensor_bytes, cudaMemcpyHostToDevice); in test_cuda_nullary()
36 cudaMemcpy(d_in2, in2.data(), tensor_bytes, cudaMemcpyHostToDevice); in test_cuda_nullary()
52 assert(cudaMemcpyAsync(new1.data(), d_in1, tensor_bytes, cudaMemcpyDeviceToHost, in test_cuda_nullary()
54 assert(cudaMemcpyAsync(new2.data(), d_in2, tensor_bytes, cudaMemcpyDeviceToHost, in test_cuda_nullary()