Searched refs:AttrTypeMap (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.cc | 35 std::unordered_map<string, const AttrTypeMap*>* OpNameToAttrTypeMap() { in OpNameToAttrTypeMap()
36 static auto* const m = new std::unordered_map<string, const AttrTypeMap*>; in OpNameToAttrTypeMap()
42 AttrTypeMap* DefaultFunctionAttrTypeMap() { in DefaultFunctionAttrTypeMap()
43 AttrTypeMap* map = new AttrTypeMap(); in DefaultFunctionAttrTypeMap()
49 const AttrTypeMap* GetDefaultFunctionAttrTypeMap() { in GetDefaultFunctionAttrTypeMap()
50 static const AttrTypeMap* map = DefaultFunctionAttrTypeMap(); in GetDefaultFunctionAttrTypeMap()
65 Status AttrTypeMapForOp(const char* op_name, const AttrTypeMap** out, in AttrTypeMapForOp()
86 std::unique_ptr<AttrTypeMap> m(new AttrTypeMap); in AttrTypeMapForOp()
194 Status AttrTypeByName(const AttrTypeMap& m, const string& attr_name, in AttrTypeByName()
|
| D | attr_builder_test.cc | 36 TEST(AttrTypeMap, Lookup) { in TEST() argument
37 const AttrTypeMap* m = nullptr; in TEST()
72 TEST(AttrTypeMap, CacheKey) { in TEST() argument
|
| D | eager_operation.h | 26 bool is_function, const tensorflow::AttrTypeMap* t) in EagerOperation()
59 const tensorflow::AttrTypeMap* AttrTypes() const { return attr_types_; } in AttrTypes()
73 const tensorflow::AttrTypeMap* attr_types_;
|
| D | attr_builder.h | 41 typedef std::unordered_map<string, uint32> AttrTypeMap; typedef
50 Status AttrTypeMapForOp(const char* op_name, const AttrTypeMap** out,
54 Status AttrTypeByName(const AttrTypeMap& m, const string& attr_name,
|
| D | execute.cc | 1057 const tensorflow::AttrTypeMap* types; in ExecuteSend()
1088 const tensorflow::AttrTypeMap* types; in ExecuteRecv()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_internal.h | 113 const tensorflow::AttrTypeMap* t, in TFE_Op()
|
| D | c_api.cc | 565 const tensorflow::AttrTypeMap* types; in TFE_NewOp()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 156 const tensorflow::AttrTypeMap* types; in ExecuteOp()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 238 const tensorflow::AttrTypeMap* attr_types; in BuildEagerOp()
|