Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h53 port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h182 virtual port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
Dstream_executor_pimpl.cc191 port::Status StreamExecutor::GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function in stream_executor::StreamExecutor
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h61 port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconv_buffer_1x1.cc247 CLKernel* ConvBuffer1x1::GetKernel(int width) { in GetKernel() function in tflite::gpu::cl::ConvBuffer1x1
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc551 TfLiteRegistration GetKernel() { in GetKernel() function
/external/tensorflow/tensorflow/core/common_runtime/
Dring_gatherer_test.cc100 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
Dring_reducer_test.cc100 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
Dhierarchical_tree_broadcaster_test.cc364 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function in tensorflow::__anonfb751b9d0111::HierarchicalTreeBroadcasterTest
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc50 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, DeviceBase* device) { in GetKernel() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc263 port::Status GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc233 port::Status GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::gpu::GpuExecutor