| /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
59 return GetKernel(node_def, device); in GetAdd()
69 return GetKernel(node_def, device); in GetDiv()
212 return GetKernel(node_def, device); in GetCollectiveReduce()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.h | 29 TfLiteRegistration GetKernel();
|
| D | delegate.cc | 69 context->ReplaceNodeSubsetsWithDelegateKernels(context, GetKernel(), in Prepare()
|
| D | kernel_test.cc | 35 context, flex::GetKernel(), size_and_nodes, delegate)); in GenericPrepare()
|
| D | kernel.cc | 560 TfLiteRegistration GetKernel() { in GetKernel() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | kernel_thunk.cc | 61 if (!executor->GetKernel(*loader_spec_, &it->second)) { in Initialize()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_reducer_test.cc | 97 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
118 return GetKernel(node_def, device_type, device); in GetAdd()
129 return GetKernel(node_def, device_type, device); in GetDiv()
381 return GetKernel(node_def, device_type, device); in GetCollectiveReduce()
|
| D | hierarchical_tree_broadcaster_test.cc | 358 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function in tensorflow::__anonf709145b0111::HierarchicalTreeBroadcasterTest
386 return GetKernel(node_def, device_type, device); in GetCollectiveBcastSend()
403 return GetKernel(node_def, device_type, device); in GetCollectiveBcastRecv()
|
| D | ring_gatherer_test.cc | 97 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
358 return GetKernel(node_def, device_type, device); in GetCollectiveGather()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 61 bool GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
|
| /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
236 return implementation_->GetKernel(spec, kernel); in GetKernel()
|
| D | stream_executor_pimpl.h | 105 bool GetKernel(const MultiKernelLoaderSpec &spec, KernelBase *kernel);
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 64 bool GetKernel(const MultiKernelLoaderSpec& spec,
|
| /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
|