Searched refs:GetDataTypeFromAttr (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 140 DataType lhs_attr = GetDataTypeFromAttr(*lhs, type_attr); in HaveSameDataType()
141 DataType rhs_attr = GetDataTypeFromAttr(*rhs, type_attr); in HaveSameDataType()
149 DataType dtype = GetDataTypeFromAttr(*node, type_attr); in HasDataType()
155 DataType dtype = GetDataTypeFromAttr(*node, type_attr); in IsCpuCompatibleDataType()
161 DataType dtype = GetDataTypeFromAttr(*node, type_attr); in IsGpuCompatibleDataType()
424 if (GetDataTypeFromAttr(node, "T") != DT_FLOAT) return false; in IsFusedBatchNormCandidate()
|
| D | arithmetic_optimizer.cc | 2188 const DataType type = GetDataTypeFromAttr(*node, "T"); in TrySimplify()
2328 const DataType type = GetDataTypeFromAttr(*node, "T"); in TrySimplify()
2376 GetDataTypeFromAttr(*node, "T") != in IsSupported()
2421 const auto type = GetDataTypeFromAttr(*node, "T"); in TrySimplify()
2929 dtype == GetDataTypeFromAttr(input, "T") && in TrySimplify()
2978 DataType dtype = GetDataTypeFromAttr(node, "T"); in CanOptimize()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.h | 276 DataType GetDataTypeFromAttr(const NodeDef& node, const string& type_attr);
|
| D | utils.cc | 312 DataType GetDataTypeFromAttr(const NodeDef& node, const string& type_attr) { in GetDataTypeFromAttr() function
|
| D | op_types.cc | 685 const auto type = GetDataTypeFromAttr(node, "T"); \
|