Searched refs:TF_AttrType (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/c/ |
D | tf_attrtype.h | 23 typedef enum TF_AttrType { enum 33 } TF_AttrType; typedef
|
D | c_api.h | 508 TF_AttrType type;
|
/external/tensorflow/tensorflow/c/eager/ |
D | c_api.h | 292 TF_CAPI_EXPORT extern TF_AttrType TFE_OpGetAttrType(TFE_Op* op, 299 TF_CAPI_EXPORT extern TF_AttrType TFE_OpNameGetAttrType(
|
D | c_api.cc | 693 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/ |
D | attr_builder.cc | 243 TF_AttrType* out, unsigned char* is_list) { in AttrTypeByName() 249 *out = static_cast<TF_AttrType>(*t & ~kIsList); in AttrTypeByName()
|
D | attr_builder_test.cc | 55 TF_AttrType t; in TEST()
|
D | attr_builder.h | 57 TF_AttrType* out, unsigned char* is_list);
|
D | BUILD | 700 # Only the TF_AttrType enum is required, so pull in just the C headers.
|
/external/tensorflow/tensorflow/python/eager/ |
D | pywrap_tfe_src.cc | 377 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/ |
D | tfe_wrapper.cc | 1501 py::enum_<TF_AttrType>(m, "TF_AttrType") in PYBIND11_MODULE()
|