Searched defs:GetKernel (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 53 port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
|
| /external/tensorflow/tensorflow/stream_executor/ |
| 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
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 60 port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_gatherer_test.cc | 101 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
|
| D | ring_reducer_test.cc | 102 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
|
| D | hierarchical_tree_broadcaster_test.cc | 372 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function in tensorflow::__anon04765bf90111::HierarchicalTreeBroadcasterTest
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 52 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, DeviceBase* device) { in GetKernel() function
|
| /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
|
| /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
|