Searched refs:eigen_gpu_device (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_gpu.cc | 63 ConcatGPUImpl<T, IntType>(c->eigen_gpu_device(), input_ptrs.data(), in ConcatGPUCall()
78 ConcatGPUSlice<T, int32>(c->eigen_gpu_device(), inputs_flat, output_flat); in ConcatGPU()
80 ConcatGPUSlice<T, int64>(c->eigen_gpu_device(), inputs_flat, output_flat); in ConcatGPU()
|
| D | ops_testutil.cc | 208 context_->eigen_gpu_device().memcpyDeviceToHost( in GetOutput()
210 context_->eigen_gpu_device().synchronize(); in GetOutput()
|
| D | bincount_op_gpu.cu.cc | 124 auto d = context->eigen_gpu_device(); in Compute()
205 auto d = context->eigen_gpu_device(); in Compute()
|
| D | gather_functor_batched_gpu.cu.h | 92 const GPUDevice& d = ctx->eigen_gpu_device();
|
| D | gather_functor_gpu.cu.h | 81 const GPUDevice& d = ctx->eigen_gpu_device();
|
| D | depthwise_conv_op_gpu.h | 663 auto device = ctx->eigen_gpu_device();
751 auto device = ctx->eigen_gpu_device();
974 auto device = ctx->eigen_gpu_device();
1587 auto device = ctx->eigen_gpu_device();
1724 auto device = ctx->eigen_gpu_device();
|
| D | dynamic_stitch_op.cc | 220 DynamicStitchGPUImpl<T>(c->eigen_gpu_device(), slice_size, first_dim_size, in Compute()
|
| D | maxpooling_op.cc | 1364 output->flat<T>().data(), nullptr, context->eigen_gpu_device(), in launch()
1385 context->eigen_gpu_device(), propagate_nans, include_batch_in_index); in launch()
1408 bottom_offset, grad_out->flat<T>().data(), context->eigen_gpu_device(), in launch()
1433 bottom_offset, grad_out->flat<T>().data(), context->eigen_gpu_device(), in launch()
|
| D | avgpooling_op.cc | 605 context->eigen_gpu_device()); // d in Compute()
|
| D | pooling_ops_common.h | 319 context->eigen_gpu_device());
|
| D | pooling_ops_3d.cc | 813 context->eigen_gpu_device()); in launch()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 270 auto device = context->eigen_gpu_device(); in NmsGpu()
382 auto device = context->eigen_gpu_device(); in CountIf()
415 auto device = context->eigen_gpu_device(); in DoNMS()
711 auto device = context->eigen_gpu_device(); in Compute()
|
| D | generate_box_proposals_op.cu.cc | 191 auto d = context->eigen_gpu_device(); in AllocateGenerationTempTensors()
228 auto d = context->eigen_gpu_device(); in AllocatePreNMSTempTensors()
|
| D | adjust_hue_op.cc | 265 GPUDevice device = context->eigen_gpu_device(); in DoCompute()
|
| D | adjust_saturation_op.cc | 233 GPUDevice device = context->eigen_gpu_device(); in DoCompute()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_util.cc | 91 sub_params_.eigen_gpu_device = nullptr;
|
| D | executor.cc | 523 p->eigen_gpu_device = nullptr; // Force allocation in ParamsButClearingEigenGPUDevice()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 596 ~Params() { delete eigen_gpu_device; } in ~Params() local
617 PerOpGpuDevice* eigen_gpu_device = nullptr; member
621 if (nullptr == eigen_gpu_device) { in ensure_eigen_gpu_device()
629 eigen_gpu_device = device->MakeGpuDevice(); in ensure_eigen_gpu_device()
1156 const Eigen::GpuDevice& eigen_gpu_device() const { in eigen_gpu_device() function
1157 return params_->eigen_gpu_device->device(); in eigen_gpu_device()
|
| D | op_kernel.cc | 336 if (params_->eigen_gpu_device != nullptr) { in OpKernelContext()
339 this, params_->eigen_gpu_device, params_->op_device_context, in OpKernelContext()
1717 return eigen_gpu_device(); in eigen_device()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 998 cc_ctx->eigen_gpu_device().memcpyHostToDevice(data, values, in set_tensor_data()
|