Home
last modified time | relevance | path

Searched refs:TF_KernelBuilder (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/
Dkernels.h65 typedef struct TF_KernelBuilder TF_KernelBuilder; typedef
102 TF_CAPI_EXPORT extern TF_KernelBuilder* TF_NewKernelBuilder(
110 TF_KernelBuilder* kernel_builder, const char* attr_name,
116 TF_KernelBuilder* kernel_builder, const char* arg_name);
120 TF_KernelBuilder* kernel_builder, int32_t priority_number);
127 TF_KernelBuilder* builder,
132 TF_CAPI_EXPORT extern void TF_DeleteKernelBuilder(TF_KernelBuilder* builder);
Dkernels.cc40 struct TF_KernelBuilder { struct
48 TF_KernelBuilder* TF_NewKernelBuilder( in TF_NewKernelBuilder() argument
53 TF_KernelBuilder* result = new TF_KernelBuilder; in TF_NewKernelBuilder()
62 void TF_DeleteKernelBuilder(TF_KernelBuilder* builder) { in TF_DeleteKernelBuilder()
78 void AddTypeConstraint(TF_KernelBuilder* kernel_builder, const char* attr_name, in AddTypeConstraint()
110 void TF_KernelBuilder_TypeConstraint(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_TypeConstraint()
118 void TF_KernelBuilder_HostMemory(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_HostMemory()
123 void TF_KernelBuilder_Priority(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_Priority()
167 explicit KernelBuilderFactory(TF_KernelBuilder* builder) in KernelBuilderFactory()
178 TF_KernelBuilder* builder_;
[all …]
Dkernels_test.cc135 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST()
212 TF_KernelBuilder* builder = TF_NewKernelBuilder( in CreateAndCallKernelWithAttr()
548 TF_KernelBuilder* builder = TF_NewKernelBuilder(op_name, device_name, nullptr, in TEST()
603 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST()
647 TF_KernelBuilder* builder = TF_NewKernelBuilder( in TEST()
678 TF_KernelBuilder* builder = TF_NewKernelBuilder( in SetupOp()
945 TF_KernelBuilder* builder = TF_NewKernelBuilder(op_name, device_name, nullptr, in TEST_F()
Dc_test.c83 TF_KernelBuilder* b = in main()
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_python_api.cc369 TF_KernelBuilder* builder = in RegisterCustomOpdefs()