Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc62 struct TypeAttrId { struct
65 explicit TypeAttrId(const string& _attr_name, int _type_index = kSingleType) in TypeAttrId() function
70 explicit TypeAttrId(DataType _fixed_type) in TypeAttrId() function
73 bool operator==(const TypeAttrId& other) const { in operator ==() argument
78 bool operator<(const TypeAttrId& other) const { in operator <() argument
84 friend H AbslHashValue(H h, const TypeAttrId& ta) { in AbslHashValue()
109 DataType GetDataType(const NodeDef& node, const TypeAttrId& type_attr) { in GetDataType()
117 if (type_attr.type_index == TypeAttrId::kSingleType) { in GetDataType()
130 bool SetDataType(NodeDef* node, const TypeAttrId& type_attr, DataType type) { in SetDataType()
135 if (type_attr.type_index == TypeAttrId::kSingleType) { in SetDataType()
[all …]