Home
last modified time | relevance | path

Searched defs:identity_nodes (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dauto_shard_dataset_op_test.cc205 std::vector<Node*>& identity_nodes) { in add_identity_nodes()
274 std::vector<Node*> identity_nodes; in TEST_F() local
324 std::vector<Node*> identity_nodes; in TEST_F() local
382 std::vector<Node*> identity_nodes; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcontrol_flow_ops_test.cc136 std::vector<Node*>& identity_nodes) { in add_identity_nodes()
201 std::vector<Node*> identity_nodes; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dbatch_dataset_op_test.cc306 std::vector<Node*>& identity_nodes) { in add_identity_nodes()
375 std::vector<Node*> identity_nodes; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dfunction_utils.cc178 gtl::InlinedVector<Node*, 8> identity_nodes(n->num_inputs(), nullptr); in RemoveListArrayConverter() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc268 std::vector<Node*> identity_nodes; in RemoveIdentityNodesForArgRetval() local