Home
last modified time | relevance | path

Searched refs:GetKernel (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc53 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, DeviceBase* device) { in GetKernel() function
69 return GetKernel(node_def, device); in GetAdd()
79 return GetKernel(node_def, device); in GetDiv()
246 return GetKernel(node_def, device); in GetCollectiveReduceOpKernel()
/external/tensorflow/tensorflow/core/common_runtime/
Dring_reducer_test.cc47 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, in GetKernel() function
68 return GetKernel(node_def, device_type, device); in GetAdd()
79 return GetKernel(node_def, device_type, device); in GetDiv()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dprogram_cache.h56 absl::Status GetKernel(uint64_t fingerprint, const std::string& function_name,
Dcl_operation.cc183 program_cache.GetKernel(kernel_fingerprint_, "main_function", &kernel_)); in RestoreDeserialized()
Dprogram_cache.cc112 absl::Status ProgramCache::GetKernel(uint64_t fingerprint, in GetKernel() function in tflite::gpu::cl::ProgramCache
/external/intel-media-driver/media_driver/agnostic/common/cm/
Dcm_task_internal.cpp760 int32_t CmTaskInternal::GetKernel( const uint32_t index, CmKernelRT* & kernel ) in GetKernel() function in CMRT_UMD::CmTaskInternal
919 CM_CHK_CMSTATUS_GOTOFINISH(GetKernel(i, cmKernel)); in VtuneInitProfilingInfo()
1019 CM_CHK_CMSTATUS_GOTOFINISH(GetKernel(i, cmKernel)); in VtuneInitProfilingInfo()
1144 this->GetKernel(i, kernelInTask); in CreateThreadSpaceData()
1548 this->GetKernel(i, tempCmKernel); in AllocateKernelSurfInfo()
1735 GetKernel(i, kernelRT); in DisplayThreadSpaceData()
Dcm_task_internal.h104 virtual int32_t GetKernel( const uint32_t index, CmKernelRT* & kernel );
Dcm_queue_rt.cpp836 task->GetKernel(i, kernel); in Enqueue_RT()
/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/service/interpreter/
Dexecutor.h59 port::Status GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel() function
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream_executor_pimpl.h107 port::Status GetKernel(const MultiKernelLoaderSpec& spec, KernelBase* kernel);
823 TF_RETURN_IF_ERROR(GetKernel(loader_spec, kernel_base.get())); in CreateTypedKernel()
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
194 return implementation_->GetKernel(spec, kernel); in GetKernel()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
Dgpu_executor.h109 port::Status GetKernel(const MultiKernelLoaderSpec& spec,
/external/chromiumos-config/go/src/go.chromium.org/chromiumos/config/go/test/artifact/
Dtko.pb.go504 func (x *Job_Test) GetKernel() *Job_Kernel { func
Dtest_result.pb.go1066 func (x *BuildMetadata) GetKernel() *BuildMetadata_Kernel { func
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc343 TF_RETURN_IF_ERROR(stream_exec->GetKernel(loader_spec, kernel_base.get())); in CreateKernel()
/external/chromiumos-config/go/src/go.chromium.org/chromiumos/config/go/build/api/
Dsystem_image.pb.go358 func (x *SystemImage_BuildMetadata_PackageSummary) GetKernel() *SystemImage_BuildMetadata_Kernel { func
/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