Searched refs:eigen_gpu_device (Results 1 – 16 of 16) 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 | 56 context_->eigen_gpu_device().memcpy(const_cast<char*>(dst.data()), in GetOutput()
58 context_->eigen_gpu_device().synchronize(); in GetOutput()
|
| D | searchsorted_op_gpu.cu.cc | 69 GetCudaLaunchConfig(values.size(), context->eigen_gpu_device()); in Compute()
89 GetCudaLaunchConfig(values.size(), context->eigen_gpu_device()); in Compute()
|
| D | gather_functor_gpu.cu.h | 79 const GPUDevice& d = ctx->eigen_gpu_device();
|
| D | adjust_saturation_op.cc | 231 GPUDevice device = context->eigen_gpu_device(); in DoCompute()
|
| D | adjust_hue_op.cc | 265 GPUDevice device = context->eigen_gpu_device(); in DoCompute()
|
| D | depthwise_conv_op_gpu.h | 638 auto device = ctx->eigen_gpu_device(); in LaunchDepthwiseConv2dGPUSmall()
761 auto device = ctx->eigen_gpu_device();
983 auto device = ctx->eigen_gpu_device();
1589 auto device = ctx->eigen_gpu_device();
1747 auto device = ctx->eigen_gpu_device();
|
| D | maxpooling_op.cc | 1308 output->flat<T>().data(), nullptr, context->eigen_gpu_device(), in launch()
1329 context->eigen_gpu_device(), propagate_nans, include_batch_in_index); in launch()
1352 bottom_offset, grad_out->flat<T>().data(), context->eigen_gpu_device(), in launch()
1377 bottom_offset, grad_out->flat<T>().data(), context->eigen_gpu_device(), in launch()
|
| D | dynamic_stitch_op.cc | 220 DynamicStitchGPUImpl<T>(c->eigen_gpu_device(), slice_size, first_dim_size, in Compute()
|
| D | avgpooling_op.cc | 548 context->eigen_gpu_device()); // d in Compute()
|
| D | pooling_ops_common.h | 279 context->eigen_gpu_device());
|
| D | pooling_ops_3d.cc | 817 context->eigen_gpu_device()); in launch()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 578 ~Params() { delete eigen_gpu_device; } in ~Params() local
599 PerOpGpuDevice* eigen_gpu_device = nullptr; member
603 if (nullptr == eigen_gpu_device) { in ensure_eigen_gpu_device()
611 eigen_gpu_device = device->MakeGpuDevice(); in ensure_eigen_gpu_device()
1114 const Eigen::GpuDevice& eigen_gpu_device() const { in eigen_gpu_device() function
1115 return params_->eigen_gpu_device->device(); in eigen_gpu_device()
|
| D | op_kernel.cc | 287 if (params_->eigen_gpu_device != nullptr) { in OpKernelContext()
290 this, params_->eigen_gpu_device, params_->op_device_context, in OpKernelContext()
1379 return eigen_gpu_device(); in eigen_device()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_util.cc | 93 sub_params_.eigen_gpu_device = nullptr;
|
| D | executor.cc | 1579 p->eigen_gpu_device = nullptr; // Force allocation in ParamsButClearingEigenGPUDevice()
|