Home
last modified time | relevance | path

Searched defs:GetKernel (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
Dhost_gpu_executor.h53 port::Status GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream_executor_internal.h174 virtual port::Status GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function
Dstream_executor_pimpl.cc192 port::Status StreamExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::StreamExecutor
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dprogram_cache.cc112 absl::Status ProgramCache::GetKernel(uint64_t fingerprint, in GetKernel() function in tflite::gpu::cl::ProgramCache
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h59 port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
/external/tensorflow/tensorflow/core/common_runtime/
Dring_reducer_test.cc47 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc53 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, DeviceBase* device) { in GetKernel() function
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc240 port::Status GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_gpu_executor.cc243 port::Status GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::gpu::GpuExecutor