Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc1460 const NodeTypeId* root_fp32 = GetTensorListFloat32NodeTypeId(*root.node); in FindFloat32TensorListOpClustersAndDenylistUnsafe() local
1462 graph_type_view_.GetNodeIndex(*root_fp32); in FindFloat32TensorListOpClustersAndDenylistUnsafe()
1464 << "Type attribute " << root_fp32->type_attr.DebugString() in FindFloat32TensorListOpClustersAndDenylistUnsafe()
1506 const NodeTypeId* root_fp32 = GetTensorListFloat32NodeTypeId(*root.node); in FindTensorListImplicitFloat32Edges() local
1507 CHECK(root_fp32) << "No float32 type attribute found for " // Crash OK in FindTensorListImplicitFloat32Edges()
1528 << root_fp32->node->op() << " node " in FindTensorListImplicitFloat32Edges()
1529 << root_fp32->node->name(); in FindTensorListImplicitFloat32Edges()
1530 implicit_fp32_edges->emplace_back(*item_fp32, *root_fp32); in FindTensorListImplicitFloat32Edges()