Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc975 const absl::flat_hash_set<const NodeDef*>& tensor_list_nodes,
982 const absl::flat_hash_set<const NodeDef*>& tensor_list_nodes,
1501 const absl::flat_hash_set<const NodeDef*>& tensor_list_nodes, in FindTensorListImplicitFloat32Edges() argument
1503 for (const NodeDef* root_node : tensor_list_nodes) { in FindTensorListImplicitFloat32Edges()
1788 const absl::flat_hash_set<const NodeDef*>& tensor_list_nodes, in ForceColorMatchBetweenTensorListOps() argument
1794 node_type_idxs.reserve(tensor_list_nodes.size()); in ForceColorMatchBetweenTensorListOps()
1795 for (const NodeDef* node : tensor_list_nodes) { in ForceColorMatchBetweenTensorListOps()