Searched defs:tensor_bytes (Results 1 – 5 of 5) sorted by relevance
114 auto tensor_bytes = input_tensors[i].tensor_data(); in Compute() local130 auto tensor_bytes = output->tensor_data(); in Compute() local
109 bool CheckSizes(size_t tensor_bytes, size_t num_values) { in CheckSizes()
112 bool CheckSizes(size_t tensor_bytes, size_t num_values) { in CheckSizes()
884 size_t tensor_bytes, bool is_gpu_to_cpu) { in CountCopies()1168 size_t tensor_bytes = (can_memcpy) ? in.TotalBytes() : proto.ByteSize(); in SendMetaData() local1199 size_t tensor_bytes = (can_memcpy) ? in.TotalBytes() : proto.ByteSize(); in SendContent() local
29 std::size_t tensor_bytes = in1.size() * sizeof(float); in test_cuda_nullary() local