Searched refs:GetDataTypeFromAttr (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 216 DataType lhs_attr = GetDataTypeFromAttr(*lhs, type_attr); in HaveSameDataType()
217 DataType rhs_attr = GetDataTypeFromAttr(*rhs, type_attr); in HaveSameDataType()
225 DataType dtype = GetDataTypeFromAttr(*node, type_attr); in HasDataType()
231 DataType dtype = GetDataTypeFromAttr(*contraction, type_attr); in IsCpuCompatibleDataType()
255 DataType dtype = GetDataTypeFromAttr(*contraction, type_attr); in IsCpuCompatibleDataType()
777 if (GetDataTypeFromAttr(*node_def, "T") != DT_FLOAT) return false; in IsCpuCompatibleDataType()
853 DataType t_dtype = GetDataTypeFromAttr(*fused_batch_norm_node_def, "T"); in IsCpuCompatibleDataType()
1697 if (GetDataTypeFromAttr(*node_def, "T") != DT_FLOAT) return false; in IsCpuCompatibleDataType()
1708 if (GetDataTypeFromAttr(*node_def, "T") != DT_FLOAT) return false; in IsCpuCompatibleDataType()
1716 if (GetDataTypeFromAttr(*relu_fanin_0_node_def, "T") != DT_FLOAT) in IsCpuCompatibleDataType()
[all …]
|
| D | arithmetic_optimizer.cc | 1889 const DataType type = GetDataTypeFromAttr(*b, "T"); in TrySimplify()
2299 const DataType type = GetDataTypeFromAttr(*node, "T"); in TrySimplify()
2441 const DataType type = GetDataTypeFromAttr(*node, "T"); in TrySimplify()
2483 GetDataTypeFromAttr(*node, "T") != in IsSupported()
2528 const auto type = GetDataTypeFromAttr(*node, "T"); in TrySimplify()
3040 dtype == GetDataTypeFromAttr(input, "T") && in TrySimplify()
3089 DataType dtype = GetDataTypeFromAttr(node, "T"); in CanOptimize()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.h | 370 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 | 776 const auto type = GetDataTypeFromAttr(node, "T"); \
|