Home
last modified time | relevance | path

Searched refs:TF_AttrBuilder (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/c/
Dc_api_experimental.h183 typedef struct TF_AttrBuilder TF_AttrBuilder; typedef
184 TF_CAPI_EXPORT extern TF_AttrBuilder* TF_NewAttrBuilder(const char* op_name);
185 TF_CAPI_EXPORT extern void TF_DeleteAttrBuilder(TF_AttrBuilder* builder);
186 TF_CAPI_EXPORT extern void TF_AttrBuilderSetType(TF_AttrBuilder* builder,
189 TF_CAPI_EXPORT extern void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder,
197 TF_AttrBuilder* builder, const char* device_type, TF_Status* status);
Dc_api_experimental.cc407 struct TF_AttrBuilder : public tensorflow::AttrBuilder { struct
415 TF_AttrBuilder* TF_NewAttrBuilder(const char* op_name) { in TF_NewAttrBuilder() argument
416 return new TF_AttrBuilder(op_name); in TF_NewAttrBuilder()
419 void TF_DeleteAttrBuilder(TF_AttrBuilder* builder) { delete builder; } in TF_DeleteAttrBuilder()
421 void TF_AttrBuilderSetType(TF_AttrBuilder* builder, const char* attr_name, in TF_AttrBuilderSetType()
427 void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder, const char* attr_name, in TF_AttrBuilderSetTypeList()
435 void TF_AttrBuilderCheckCanRunOnDevice(TF_AttrBuilder* builder, in TF_AttrBuilderCheckCanRunOnDevice()