Home
last modified time | relevance | path

Searched defs:kernel_name (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dlog_memory.proto25 string kernel_name = 2; field
46 string kernel_name = 2; field
Dlog_memory.cc48 void LogMemory::RecordTensorAllocation(const string& kernel_name, in RecordTensorAllocation()
66 void LogMemory::RecordTensorOutput(const string& kernel_name, in RecordTensorOutput()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_thunk.h57 const string& kernel_name() const { return kernel_name_; } in kernel_name() function
Dkernel_thunk.cc31 const string& kernel_name, in KernelThunk()
Dir_emitter_unnested.cc193 string kernel_name = ir_emitter_context_->name_uniquer()->GetUniqueName( in BuildKernelPrototype() local
/external/gemmlowp/meta/generators/
Dcommon.py89 def __init__(self, emitter, kernel_name, output_stream_name): argument
118 def __init__(self, emitter, kernel_name): argument
Dtransform_kernels_common.py308 def __init__(self, cc_emitter, kernel_name, asm_emitter, transformation): argument
Dquantized_mul_kernels_common.py540 def __init__(self, cc_emitter, kernel_name, output_stream_name, asm_emitter, argument
/external/ImageMagick/MagickCore/
Dopencl.h35 *kernel_name; member
Dopencl.c704 const char *kernel_name) in AcquireOpenCLKernel()
/external/libxcam/tests/
Dtest-binary-kernel.cpp53 char *kernel_name = NULL; in main() local
/external/tensorflow/tensorflow/c/
Dkernels_test.cc89 const char* kernel_name = "SomeKernelName"; in TEST() local
141 const char* kernel_name = "InputOutputCounterKernel"; in TEST() local
/external/mesa3d/src/gallium/state_trackers/clover/llvm/codegen/
Dcommon.cpp70 make_kernel_args(const Module &mod, const std::string &kernel_name, in make_kernel_args()
/external/tensorflow/tensorflow/stream_executor/
Dkernel_spec.cc50 absl::string_view kernel_name, in CudaPtxInMemory()
64 absl::string_view kernel_name, bool ptx_compressed) in CudaPtxInMemory()
/external/libxcam/modules/ocl/
Dcl_kernel.h45 const char *kernel_name; member
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_support_library.h273 absl::string_view kernel_name, llvm::Value* arg0, llvm::Value* arg1, in EmitAndCallOutlinedKernel()
286 absl::string_view kernel_name, llvm::Value* arg0, llvm::Value* arg1, in EmitAndCallOutlinedKernel()
Dkernel_support_library.cc74 absl::string_view kernel_name, in EmitAndCallOutlinedKernel()
/external/autotest/tko/
Dcompose_query.cgi178 def map_kernel_base(kernel_name): argument
/external/tensorflow/tensorflow/core/platform/default/
Ddevice_tracer.cc74 const char* kernel_name; member
111 size_t StartKernel(const char* kernel_name, CUcontext context, in StartKernel()
/external/gemmlowp/standalone/
Dneon-gemm-kernel-benchmark.cc4958 void test_kernel(int depth, const char* kernel_name) { in test_kernel()
5104 void benchmark_and_print_results(const char* kernel_name) { in benchmark_and_print_results()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc1152 const char* kernel_name = "testkernel"; in PopulateDeviceDescription() local
Dcuda_driver.cc732 const char* kernel_name, in GetModuleFunction()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc573 const char* kernel_name, in GetModuleFunction()