Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
H A Dhost_gpu_executor.h53 port::Status GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream_executor_internal.h174 virtual port::Status GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function
H A Dstream_executor_pimpl.cc192 port::Status StreamExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::StreamExecutor
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dprogram_cache.cc112 absl::Status ProgramCache::GetKernel(uint64_t fingerprint, in GetKernel() function in tflite::gpu::cl::ProgramCache
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutor.h59 port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
/aosp_15_r20/test/vts-testcase/security/avb/
Dvts_gki_compliance_test.cpp96 std::vector<uint8_t> GetKernel() const { in GetKernel() function in __anon738bffe90111::GkiBootImage
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dring_reducer_test.cc47 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcollective_nccl_test.cc53 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, DeviceBase* device) { in GetKernel() function
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc240 port::Status GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::gpu::GpuExecutor
/aosp_15_r20/external/intel-media-driver/media_driver/agnostic/common/cm/
H A Dcm_task_internal.cpp760 int32_t CmTaskInternal::GetKernel( const uint32_t index, CmKernelRT* & kernel ) in GetKernel() function in CMRT_UMD::CmTaskInternal
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc243 port::Status GpuExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel() function in stream_executor::gpu::GpuExecutor