Home
last modified time | relevance | path

Searched refs:CreateKernel (Results 1 – 22 of 22) 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.cc84 Status XlaKernelCreator::CreateKernel( in CreateKernel() function in tensorflow::XlaKernelCreator
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dtfrt_op_kernel_test.cc153 tfrt_forwarding_kernel_factories->CreateKernel("TestKernelFloatInt", in TEST()
180 tfrt_forwarding_kernel_factories->CreateKernel("TestKernel2ndConstraint", in TEST()
201 tfrt_forwarding_kernel_factories->CreateKernel("TestKernelIntInt", &ctx); in TEST()
Dkernel_fallback_execute.cc83 tfrt_forwarding_kernel_factories->CreateKernel( in KernelFallbackExecute()
Dtfrt_op_kernel.h285 std::unique_ptr<TFRTOpKernel> CreateKernel(
Dop_kernel_runner.cc76 TF_RETURN_IF_ERROR(flr->CreateKernel(props, &k)); in CreateOpKernel()
Dtfrt_op_kernel.cc294 std::unique_ptr<TFRTOpKernel> TFRTOpKernelFactories::CreateKernel( in CreateKernel() function in tensorflow::TFRTOpKernelFactories
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc178 Status CreateKernel(const std::shared_ptr<const NodeProperties>& props,
259 Status FunctionLibraryRuntimeOverlay::CreateKernel( in CreateKernel() function in tensorflow::FunctionLibraryRuntimeOverlay
347 Status CreateKernel(const std::shared_ptr<const NodeProperties>& props,
434 Status CreateKernel(const std::shared_ptr<const NodeProperties>& props,
490 return CreateKernel(props, kernel); in FunctionLibraryRuntimeImpl()
603 Status FunctionLibraryRuntimeImpl::CreateKernel( in CreateKernel() function in tensorflow::FunctionLibraryRuntimeImpl
605 return CreateKernel(props, this, kernel); in CreateKernel()
608 Status FunctionLibraryRuntimeImpl::CreateKernel( in CreateKernel() function in tensorflow::FunctionLibraryRuntimeImpl
618 s = custom_kernel_creator->CreateKernel(flr, props, &ret); in CreateKernel()
1027 return CreateKernel(props, flr, kernel); in CreateItem()
Ddirect_session.cc1379 return lib->CreateKernel(props, kernel); in CreateExecutors()
1382 return lib->CreateKernel(props, kernel); in CreateExecutors()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.h81 StatusOr<std::unique_ptr<se::KernelBase>> CreateKernel(
Dkernel_thunk.cc61 CreateKernel(kernel_name_, args_.size(), executable.text(), in Initialize()
Dstream_executor_util.cc329 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.h865 virtual Status CreateKernel(
970 virtual Status CreateKernel(
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc138 Status s = flib_->CreateKernel(n->properties(), &op_kernel_raw); in Compile()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc113 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.cc571 status = params_.function_library->CreateKernel(props, &op_kernel_raw); in LegalizeOp()