Home
last modified time | relevance | path

Searched refs:cudaFree (Results 1 – 16 of 16) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_cuda.cu64 cudaFree(d_in1); in test_cuda_nullary()
65 cudaFree(d_in2); in test_cuda_nullary()
111 cudaFree(d_in1); in test_cuda_elementwise_small()
112 cudaFree(d_in2); in test_cuda_elementwise_small()
113 cudaFree(d_out); in test_cuda_elementwise_small()
165 cudaFree(d_in1); in test_cuda_elementwise()
166 cudaFree(d_in2); in test_cuda_elementwise()
167 cudaFree(d_in3); in test_cuda_elementwise()
168 cudaFree(d_out); in test_cuda_elementwise()
204 cudaFree(d_in1); in test_cuda_props()
[all …]
Dcxx11_tensor_contract_cuda.cu84 cudaFree((void*)d_t_left); in test_cuda_contraction()
85 cudaFree((void*)d_t_right); in test_cuda_contraction()
86 cudaFree((void*)d_t_result); in test_cuda_contraction()
142 cudaFree((void*)d_t_left); in test_scalar()
143 cudaFree((void*)d_t_right); in test_scalar()
144 cudaFree((void*)d_t_result); in test_scalar()
Dcxx11_tensor_argmax_cuda.cu63 cudaFree(d_in); in test_cuda_simple_argmax()
64 cudaFree(d_out_max); in test_cuda_simple_argmax()
65 cudaFree(d_out_min); in test_cuda_simple_argmax()
152 cudaFree(d_in); in test_cuda_argmax_dim()
153 cudaFree(d_out); in test_cuda_argmax_dim()
241 cudaFree(d_in); in test_cuda_argmin_dim()
242 cudaFree(d_out); in test_cuda_argmin_dim()
Dcxx11_tensor_complex_cwise_ops_cuda.cu87 cudaFree(d_in1); in test_cuda_complex_cwise_ops()
88 cudaFree(d_in2); in test_cuda_complex_cwise_ops()
89 cudaFree(d_out); in test_cuda_complex_cwise_ops()
Dcxx11_tensor_scan_cuda.cu70 cudaFree((void*)d_t_input); in test_cuda_cumsum()
71 cudaFree((void*)d_t_result); in test_cuda_cumsum()
Dcxx11_tensor_complex_cuda.cu68 cudaFree(d_in1); in test_cuda_nullary()
69 cudaFree(d_in2); in test_cuda_nullary()
70 cudaFree(d_out2); in test_cuda_nullary()
Dcxx11_tensor_device.cu84 assert(cudaFree(kernel_1d_) == cudaSuccess); in ~GPUContext()
85 assert(cudaFree(kernel_2d_) == cudaSuccess); in ~GPUContext()
86 assert(cudaFree(kernel_3d_) == cudaSuccess); in ~GPUContext()
/external/eigen/test/
Dcuda_common.h60 cudaFree(d_in); in run_on_cuda()
61 cudaFree(d_out); in run_on_cuda()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtensorrt_test.cc130 ASSERT_EQ(0, cudaFree(buffers[input_index])); in Execute()
131 ASSERT_EQ(0, cudaFree(buffers[output_index])); in Execute()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_managed_allocator.cc39 CHECK_EQ(cudaFree(ptr), cudaSuccess); in DeallocateRaw()
Dgpu_device.cc1039 err = cudaFree(nullptr); in CreateDevices()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_allocator.cc69 void TRTCudaAllocator::free(void* memory) { cudaFree(memory); } in free()
/external/tensorflow/tensorflow/core/util/
Dcuda_kernel_helper_test.cu.cc146 cudaFree(outbuf); in TearDown()
288 cudaFree(failure_count); in TEST()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudart_stub.cc51 cudaError_t CUDART_CB cudaFree(void* devPtr) { in cudaFree() function
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h154 err = cudaFree(buffer); in initializeDeviceProp()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1181 ASSERT_EQ(0, cudaFree(buffers[i])); in BuildAndRun()