Searched refs:AttrTypeMap (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.cc | 35 tensorflow::gtl::FlatMap<string, const AttrTypeMap*>* OpNameToAttrTypeMap() { in OpNameToAttrTypeMap()
37 new tensorflow::gtl::FlatMap<string, const AttrTypeMap*>; in OpNameToAttrTypeMap()
43 AttrTypeMap* DefaultFunctionAttrTypeMap() { in DefaultFunctionAttrTypeMap()
44 AttrTypeMap* map = new AttrTypeMap(); in DefaultFunctionAttrTypeMap()
50 const AttrTypeMap* GetDefaultFunctionAttrTypeMap() { in GetDefaultFunctionAttrTypeMap()
51 static const AttrTypeMap* map = DefaultFunctionAttrTypeMap(); in GetDefaultFunctionAttrTypeMap()
66 Status AttrTypeMapForOp(const char* op_name, const AttrTypeMap** out, in AttrTypeMapForOp()
98 std::unique_ptr<AttrTypeMap> m(new AttrTypeMap); in AttrTypeMapForOp()
242 Status AttrTypeByName(const AttrTypeMap& m, const string& attr_name, in AttrTypeByName()
|
| D | attr_builder_test.cc | 37 TEST(AttrTypeMap, Lookup) { in TEST() argument
38 const AttrTypeMap* m = nullptr; in TEST()
73 TEST(AttrTypeMap, CacheKey) { in TEST() argument
|
| D | attr_builder.h | 43 typedef std::unordered_map<string, uint32> AttrTypeMap; typedef
52 Status AttrTypeMapForOp(const char* op_name, const AttrTypeMap** out,
56 Status AttrTypeByName(const AttrTypeMap& m, const string& attr_name,
|
| D | eager_operation.h | 222 const AttrTypeMap* attr_types_;
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 696 const tensorflow::AttrTypeMap* attr_types_; in TFE_OpGetAttrType()
|