Searched refs:KernelDefBuilder (Results 1 – 5 of 5) sorted by relevance
22 KernelDefBuilder::KernelDefBuilder(const char* op_name) { in KernelDefBuilder() function in tensorflow::KernelDefBuilder27 KernelDefBuilder::~KernelDefBuilder() { in ~KernelDefBuilder()31 KernelDefBuilder& KernelDefBuilder::Device(const char* device_type) { in Device()37 KernelDefBuilder& KernelDefBuilder::AttrConstraint<int64>( in AttrConstraint()50 KernelDefBuilder& KernelDefBuilder::AttrConstraint<int64>(const char* attr_name, in AttrConstraint()58 KernelDefBuilder& KernelDefBuilder::AttrConstraint<string>( in AttrConstraint()70 KernelDefBuilder& KernelDefBuilder::AttrConstraint<string>( in AttrConstraint()78 KernelDefBuilder& KernelDefBuilder::AttrConstraint<const char*>( in AttrConstraint()90 KernelDefBuilder& KernelDefBuilder::AttrConstraint<const char*>( in AttrConstraint()98 KernelDefBuilder& KernelDefBuilder::AttrConstraint<bool>(const char* attr_name, in AttrConstraint()[all …]
30 class KernelDefBuilder {34 explicit KernelDefBuilder(const char* op_name);35 ~KernelDefBuilder();39 KernelDefBuilder& Device(const char* device_type);46 KernelDefBuilder& AttrConstraint(const char* attr_name,51 KernelDefBuilder& AttrConstraint(const char* attr_name, T allowed);57 KernelDefBuilder& TypeConstraint(const char* attr_name,61 KernelDefBuilder& TypeConstraint(const char* attr_name, DataType allowed);66 KernelDefBuilder& TypeConstraint(const char* attr_name);72 KernelDefBuilder& HostMemory(const char* arg_name);[all …]
27 const KernelDef* def = KernelDefBuilder("A").Device(DEVICE_CPU).Build(); in TEST()36 const KernelDef* def = KernelDefBuilder("B") in TEST()49 def = KernelDefBuilder("C") in TEST()63 def = KernelDefBuilder("D") in TEST()77 const KernelDef* def = KernelDefBuilder("B") in TEST()94 def = KernelDefBuilder("C") in TEST()118 const KernelDef* def = KernelDefBuilder("B") in TEST()135 def = KernelDefBuilder("C") in TEST()159 const KernelDef* def = KernelDefBuilder("E") in TEST()
1420 class Name : public KernelDefBuilder {1422 explicit Name(const char* op) : KernelDefBuilder(op) {} in Name()
41 ::tensorflow::KernelDefBuilder* cc_builder;54 result->cc_builder = new ::tensorflow::KernelDefBuilder(op_name); in TF_NewKernelBuilder()