Home
last modified time | relevance | path

Searched refs:CreateKernel (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/image/
Dscale_and_translate_op_test.cc56 std::unique_ptr<const DynamicKernel> CreateKernel(const KernelType& kernel) { in CreateKernel() function
64 return CreateKernel(functor::CreateLanczos1Kernel()); in Create()
66 return CreateKernel(functor::CreateLanczos3Kernel()); in Create()
68 return CreateKernel(functor::CreateLanczos5Kernel()); in Create()
70 return CreateKernel(functor::CreateGaussianKernel()); in Create()
72 return CreateKernel(functor::CreateBoxKernel()); in Create()
74 return CreateKernel(functor::CreateTriangleKernel()); in Create()
76 return CreateKernel(functor::CreateKeysCubicKernel()); in Create()
78 return CreateKernel(functor::CreateMitchellCubicKernel()); in Create()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_kernel_creator_test.cc110 Status status = xla_kernel_creator.CreateKernel(flr_, callsite, &kernel_); in TEST_F()
133 xla_kernel_creator.CreateKernel(flr_, ToNodeProperties(R"proto( in TEST_F()
150 xla_kernel_creator.CreateKernel(flr_, ToNodeProperties(R"proto( in TEST_F()
Dxla_kernel_creator.h37 Status CreateKernel(FunctionLibraryRuntime* flr,
Dxla_kernel_creator.cc172 Status XlaKernelCreator::CreateKernel( in CreateKernel() function in tensorflow::XlaKernelCreator
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc175 Status CreateKernel(const std::shared_ptr<const NodeProperties>& props,
256 Status FunctionLibraryRuntimeOverlay::CreateKernel( in CreateKernel() function in tensorflow::FunctionLibraryRuntimeOverlay
344 Status CreateKernel(const std::shared_ptr<const NodeProperties>& props,
431 Status CreateKernel(const std::shared_ptr<const NodeProperties>& props,
485 return CreateKernel(props, kernel); in FunctionLibraryRuntimeImpl()
598 Status FunctionLibraryRuntimeImpl::CreateKernel( in CreateKernel() function in tensorflow::FunctionLibraryRuntimeImpl
600 return CreateKernel(props, this, kernel); in CreateKernel()
603 Status FunctionLibraryRuntimeImpl::CreateKernel( in CreateKernel() function in tensorflow::FunctionLibraryRuntimeImpl
613 s = custom_kernel_creator->CreateKernel(this, props, &ret); in CreateKernel()
939 return CreateKernel(props, flr, kernel); in CreateItem()
Ddirect_session.cc1384 return lib->CreateKernel(props, kernel); in CreateExecutors()
1387 return lib->CreateKernel(props, kernel); in CreateExecutors()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.h69 StatusOr<std::unique_ptr<se::KernelBase>> CreateKernel(
Dkernel_thunk.cc56 CreateKernel(kernel_name_, args_.size(), executable.text(), in Initialize()
Dstream_executor_util.cc194 StatusOr<std::unique_ptr<se::KernelBase>> CreateKernel( in CreateKernel() function
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc256 return lib->CreateKernel(props, kernel); in InitItem()
259 return lib->CreateKernel(props, kernel); in InitItem()
/external/tensorflow/tensorflow/core/framework/
Dfunction.h838 virtual Status CreateKernel(
944 virtual Status CreateKernel(
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc139 Status s = flib_->CreateKernel(n->properties(), &op_kernel_raw); in Compile()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc112 TF_RETURN_IF_ERROR(flr_->CreateKernel(props, &k)); in Init()
/external/angle/src/libGLESv2/
Dcl_stubs_autogen.h226 cl_kernel CreateKernel(cl_program program, const char *kernel_name, cl_int &errorCode);
Dentry_points_cl_autogen.cpp550 CL_EVENT(CreateKernel, in clCreateKernel()
555 ANGLE_CL_VALIDATE_ERRCODE_RET(CreateKernel, program, kernel_name); in clCreateKernel()
558 cl_kernel object = CreateKernel(program, kernel_name, errorCode); in clCreateKernel()
Dcl_stubs.cpp532 cl_kernel CreateKernel(cl_program program, const char *kernel_name, cl_int &errorCode) in CreateKernel() function
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc438 status = params_.function_library->CreateKernel(props, &op_kernel_raw); in LegalizeOp()