Searched refs:dev_buf (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tile_functor_gpu.h | 70 auto dev_buf = d.allocate(num_bytes); in TileSimple() local
73 d.memcpyHostToDevice(dev_buf, host_buf.data(), num_bytes); in TileSimple()
81 reinterpret_cast<const int32*>(dev_buf), ndims, q)); in TileSimple()
83 d.deallocate(dev_buf); in TileSimple()
|
| D | transpose_functor_gpu.cu.cc | 76 auto dev_buf = d.allocate(num_bytes); in TransposeSimple() local
79 d.memcpyHostToDevice(dev_buf, host_buf.data(), num_bytes); in TransposeSimple()
87 reinterpret_cast<const int32*>(dev_buf), ndims, q)); in TransposeSimple()
89 d.deallocate(dev_buf); in TransposeSimple()
|
| /external/strace/ |
| D | dm.c | 219 uint64_t dev_buf; in dm_decode_dm_target_deps() local
231 space = (ioc->data_size - offset_end) / sizeof(dev_buf); in dm_decode_dm_target_deps()
239 print_array(tcp, addr + offset_end, s.count, &dev_buf, sizeof(dev_buf), in dm_decode_dm_target_deps()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.cc | 224 string dev_buf; in CompleteGroupLocal() local
226 strings::StrAppend(&dev_buf, ",", d.first); in CompleteGroupLocal()
230 << " devices)=(" << dev_buf << ") (number of" in CompleteGroupLocal()
|