Home
last modified time | relevance | path

Searched refs:TF_AttrType (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/c/
Dtf_attrtype.h23 typedef enum TF_AttrType { enum
33 } TF_AttrType; typedef
Dc_api.h508 TF_AttrType type;
/external/tensorflow/tensorflow/c/eager/
Dc_api.h292 TF_CAPI_EXPORT extern TF_AttrType TFE_OpGetAttrType(TFE_Op* op,
299 TF_CAPI_EXPORT extern TF_AttrType TFE_OpNameGetAttrType(
Dc_api.cc693 TF_AttrType TFE_OpGetAttrType(TFE_Op* op, const char* attr_name, in TFE_OpGetAttrType()
695 TF_AttrType ret = TF_ATTR_INT; in TFE_OpGetAttrType()
708 TF_AttrType TFE_OpNameGetAttrType(TFE_Context* ctx, in TFE_OpNameGetAttrType()
712 TF_AttrType ret; in TFE_OpNameGetAttrType()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder.cc243 TF_AttrType* out, unsigned char* is_list) { in AttrTypeByName()
249 *out = static_cast<TF_AttrType>(*t & ~kIsList); in AttrTypeByName()
Dattr_builder_test.cc55 TF_AttrType t; in TEST()
Dattr_builder.h57 TF_AttrType* out, unsigned char* is_list);
DBUILD700 # Only the TF_AttrType enum is required, so pull in just the C headers.
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc377 TF_AttrType type, in SetOpAttrList()
540 const char* key, TF_AttrType type, in SetOpAttrListDefault()
635 TF_AttrType type, in SetOpAttrScalar()
763 const TF_AttrType type = TFE_OpGetAttrType(op, key, &is_list, out_status); in SetOpAttrs()
785 const TF_AttrType type = TFE_OpGetAttrType(op, attr_name, &is_list, status); in SetOpAttrWithDefaults()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc1501 py::enum_<TF_AttrType>(m, "TF_AttrType") in PYBIND11_MODULE()