Searched refs:devptr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
| D | device_tracer_test.cc | 306 void* devptr = 0; in TEST_F() local
308 ASSERT_EQ(cudaSuccess, cudaMalloc(&devptr, size_in_bytes)); in TEST_F()
309 VLOG(3) << "Allocated device memory, addr: " << devptr; in TEST_F()
310 ASSERT_EQ(cudaSuccess, cudaMemset(devptr, test_value, size_in_bytes)); in TEST_F()
313 cudaMemcpy(buffer, devptr, size_in_bytes, cudaMemcpyDeviceToHost)); in TEST_F()
314 VLOG(3) << "Free device memory, addr: " << devptr; in TEST_F()
315 ASSERT_EQ(cudaSuccess, cudaFree(devptr)); in TEST_F()
382 if (addr == reinterpret_cast<size_t>(devptr) && in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_int8_calibrator.cc | 62 auto devptr = dev_buffers_.find(it.first); in setBatch() local
63 if (devptr == dev_buffers_.end()) { in setBatch()
67 const auto& d = devptr->second; in setBatch()
|