Searched refs:while_builder (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util_test.cc | 221 NodeBuilder while_builder(kFunctionalWhileNodeName, "While", in TEST_F() local
223 while_builder.Input({input_node, input_node}) in TEST_F()
226 builder.opts().FinalizeBuilder(&while_builder); in TEST_F()
|
| D | extract_outside_compilation_pass.cc | 1713 NodeDefBuilder while_builder(absl::StrCat("oc_while_", while_node_name), in BuildHostGraphForWhileNode() local
1715 while_builder.Attr("T", std::vector<DataType>{DT_STRING}); in BuildHostGraphForWhileNode()
1721 while_builder.Attr("cond", func); in BuildHostGraphForWhileNode()
1723 while_builder.Attr("body", func); in BuildHostGraphForWhileNode()
1724 while_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForWhileNode()
1725 while_builder.Attr(xla_cluster_attr_name, xla_cluster_name); in BuildHostGraphForWhileNode()
1726 while_builder.Attr(outside_compilation_attr_name, outside_compilation_name); in BuildHostGraphForWhileNode()
1729 while_builder.Attr("parallel_iterations", 1); in BuildHostGraphForWhileNode()
1732 while_builder.Input(while_inputs); in BuildHostGraphForWhileNode()
1734 TF_RETURN_IF_ERROR(while_builder.Finalize(&while_def)); in BuildHostGraphForWhileNode()
|