Searched refs:KernelDefBuilder (Results 1 – 5 of 5) sorted by relevance
23 KernelDefBuilder::KernelDefBuilder(const char* op_name) { in KernelDefBuilder() function in tensorflow::KernelDefBuilder28 KernelDefBuilder::~KernelDefBuilder() { in ~KernelDefBuilder()32 KernelDefBuilder& KernelDefBuilder::Device(const char* device_type) { in Device()37 KernelDefBuilder& KernelDefBuilder::TypeConstraint( in TypeConstraint()48 KernelDefBuilder& KernelDefBuilder::TypeConstraint(const char* attr_name, in TypeConstraint()56 KernelDefBuilder& KernelDefBuilder::HostMemory(const char* arg_name) { in HostMemory()61 KernelDefBuilder& KernelDefBuilder::Label(const char* label) { in Label()69 KernelDefBuilder& KernelDefBuilder::Priority(int32 priority) { in Priority()74 const KernelDef* KernelDefBuilder::Build() { in Build()
30 class KernelDefBuilder {34 explicit KernelDefBuilder(const char* op_name);35 ~KernelDefBuilder();39 KernelDefBuilder& Device(const char* device_type);46 KernelDefBuilder& TypeConstraint(const char* attr_name,50 KernelDefBuilder& TypeConstraint(const char* attr_name, DataType allowed);55 KernelDefBuilder& TypeConstraint(const char* attr_name);61 KernelDefBuilder& HostMemory(const char* arg_name);65 KernelDefBuilder& Label(const char* label);68 KernelDefBuilder& Priority(int32 priority);[all …]
26 const KernelDef* def = KernelDefBuilder("A").Device(DEVICE_CPU).Build(); in TEST()35 const KernelDef* def = KernelDefBuilder("B") in TEST()48 def = KernelDefBuilder("C") in TEST()62 def = KernelDefBuilder("D") in TEST()76 const KernelDef* def = KernelDefBuilder("E") in TEST()
1346 class Name : public KernelDefBuilder {1359 : KernelDefBuilder(SHOULD_REGISTER_OP(op) ? op : "_no_register") {} in Name()1364 class Name : public KernelDefBuilder {1368 explicit Name(const char* op) : KernelDefBuilder(op) {} in Name()
29 ::tensorflow::KernelDefBuilder* cc_builder;42 result->cc_builder = new ::tensorflow::KernelDefBuilder(op_name); in TF_NewKernelBuilder()