| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 52 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, DeviceBase* device) { in GetKernel() function
68 return GetKernel(node_def, device); in GetAdd()
78 return GetKernel(node_def, device); in GetDiv()
244 return GetKernel(node_def, device); in GetCollectiveReduceOpKernel()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_reducer_test.cc | 102 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
123 return GetKernel(node_def, device_type, device); in GetAdd()
134 return GetKernel(node_def, device_type, device); in GetDiv()
392 return GetKernel(node_def, device_type, device); in GetCollectiveReduce()
|
| D | hierarchical_tree_broadcaster_test.cc | 372 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function in tensorflow::__anon04765bf90111::HierarchicalTreeBroadcasterTest
400 return GetKernel(node_def, device_type, device); in GetCollectiveBcastSend()
417 return GetKernel(node_def, device_type, device); in GetCollectiveBcastRecv()
|
| D | ring_gatherer_test.cc | 101 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
367 return GetKernel(node_def, device_type, device); in GetCollectiveGather()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 53 port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 60 port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 105 port::Status GetKernel(const MultiKernelLoaderSpec &spec, KernelBase *kernel);
805 TF_RETURN_IF_ERROR(GetKernel(loader_spec, kernel_base.get())); in CreateTypedKernel()
|
| D | stream_executor_internal.h | 181 virtual port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
|
| D | stream_executor_pimpl.cc | 192 port::Status StreamExecutor::GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function in stream_executor::StreamExecutor
194 return implementation_->GetKernel(spec, kernel); in GetKernel()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 64 port::Status GetKernel(const MultiKernelLoaderSpec& spec,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.cc | 206 TF_RETURN_IF_ERROR(stream_exec->GetKernel(loader_spec, kernel_base.get())); in CreateKernel()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 238 port::Status GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 263 port::Status GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::gpu::GpuExecutor
|