| /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_t>(c->eigen_gpu_device(), inputs_flat, in ConcatGPU()
|
| D | sequence_ops_gpu.cu.cc | 47 const GPUDevice& device = context->eigen_gpu_device(); in operator ()()
|
| D | ops_testutil.cc | 208 context_->eigen_gpu_device().memcpyDeviceToHost( in GetOutput()
210 context_->eigen_gpu_device().synchronize(); in GetOutput()
|
| D | sparse_to_dense_op_gpu.cu.cc | 116 const Eigen::GpuDevice& d = c->eigen_gpu_device(); in LaunchComputeKernels()
147 const Eigen::GpuDevice& d = c->eigen_gpu_device(); in operator ()()
|
| D | bincount_op_gpu.cu.cc | 135 auto d = context->eigen_gpu_device(); in Compute()
216 auto d = context->eigen_gpu_device(); in Compute()
|
| D | tensor_to_hash_bucket_op_gpu.cu.cc | 91 const Eigen::GpuDevice& d = c->eigen_gpu_device(); in operator ()()
|
| D | sparse_slice_grad_op_gpu.cu.cc | 143 const GPUDevice& device = ctx->eigen_gpu_device(); in operator ()()
|
| D | sparse_reorder_op_gpu.cu.cc | 72 const Eigen::GpuDevice& d = c->eigen_gpu_device(); in operator ()()
|
| D | gather_functor_gpu.cu.h | 130 const GPUDevice& d = ctx->eigen_gpu_device();
|
| D | gather_functor_batched_gpu.cu.h | 146 const GPUDevice& d = ctx->eigen_gpu_device();
|
| D | sparse_concat_op_gpu.cu.cc | 193 const GPUDevice& device = context->eigen_gpu_device(); in operator ()()
|
| D | depthwise_conv_op_gpu.h | 665 auto device = ctx->eigen_gpu_device();
749 auto device = ctx->eigen_gpu_device();
972 auto device = ctx->eigen_gpu_device();
1548 auto device = ctx->eigen_gpu_device();
1669 auto device = ctx->eigen_gpu_device();
|
| D | segment_reduction_ops_gpu.cu.h | 529 const GPUDevice& device = ctx->eigen_gpu_device();
549 const GPUDevice& device = ctx->eigen_gpu_device();
942 const GPUDevice& device = context->eigen_gpu_device();
|
| D | dynamic_stitch_op.cc | 221 DynamicStitchGPUImpl<T>(c->eigen_gpu_device(), slice_size, first_dim_size, in Compute()
|
| D | sparse_split_op_gpu.cu.cc | 227 const GPUDevice& device = context->eigen_gpu_device(); in operator ()()
|
| D | unique_op_gpu.cu.h | 332 const GPUDevice& device = context->eigen_gpu_device(); in ComputeAsync()
|
| D | maxpooling_op.cc | 1438 output->flat<T>().data(), nullptr, context->eigen_gpu_device(), in launch()
1459 context->eigen_gpu_device(), propagate_nans, include_batch_in_index); in launch()
1483 context->eigen_gpu_device(), include_batch_in_index); in launch()
1508 context->eigen_gpu_device(), include_batch_in_index); in launch()
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | op_kernel_runner_test.cc | 57 ASSERT_THAT(params.eigen_gpu_device, ::testing::NotNull()); in TEST()
74 EXPECT_THAT(run_state.params.eigen_gpu_device, IsNull()); in TEST()
|
| D | op_kernel_runner.h | 127 params.eigen_gpu_device = nullptr; in OpKernelRunState()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 563 ~Params() { delete eigen_gpu_device; } in ~Params() local
590 PerOpGpuDevice* eigen_gpu_device = nullptr; member
594 if (nullptr == eigen_gpu_device) { in ensure_eigen_gpu_device()
602 eigen_gpu_device = device->MakeGpuDevice(); in ensure_eigen_gpu_device()
1112 const Eigen::GpuDevice& eigen_gpu_device() const { in eigen_gpu_device() function
1113 return params_->eigen_gpu_device->device(); in eigen_gpu_device()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 278 auto device = context->eigen_gpu_device(); in CountIf()
322 auto device = context->eigen_gpu_device(); in DoNMS()
640 auto device = context->eigen_gpu_device(); in Compute()
685 auto device = context->eigen_gpu_device(); in NmsGpu()
|
| 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 | 92 sub_params_.eigen_gpu_device = nullptr;
|