Searched defs:GetKernel (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 53 bool GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_internal.h | 182 virtual bool GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
|
| D | stream_executor_pimpl.cc | 234 bool StreamExecutor::GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function in stream_executor::StreamExecutor
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 61 bool GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer_test.cc | 43 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, DeviceBase* device) { in GetKernel() function
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 560 TfLiteRegistration GetKernel() { in GetKernel() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_gatherer_test.cc | 97 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
|
| D | ring_reducer_test.cc | 97 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
|
| D | hierarchical_tree_broadcaster_test.cc | 358 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function in tensorflow::__anonf709145b0111::HierarchicalTreeBroadcasterTest
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 221 bool GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 267 bool GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::gpu::GpuExecutor
|