Searched refs:HasDataType (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 223 bool HasDataType(const NodeDef* node, const DataType& expected, in HasDataType() function
554 !HasDataType(node_def, DT_FLOAT, "U")) in IsCpuCompatibleDataType()
682 if (!HasDataType(node_def, DT_FLOAT) && !HasDataType(node_def, DT_BFLOAT16)) in IsCpuCompatibleDataType()
686 if (!HasDataType(node_def, DT_FLOAT)) return false; in IsCpuCompatibleDataType()
735 if (!HasDataType(node_def, DT_FLOAT) && !HasDataType(node_def, DT_BFLOAT16)) in IsCpuCompatibleDataType()
739 if (!HasDataType(node_def, DT_FLOAT)) return false; in IsCpuCompatibleDataType()
894 !HasDataType(fused_batch_norm_node_def, DT_FLOAT, "U")) in IsCpuCompatibleDataType()
|