Searched refs:func_name_attrs (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.h | 100 const NameAttrList& func_name_attrs, Node* node, in XlaClusterInfo()
103 func_name_attrs(func_name_attrs), in XlaClusterInfo()
109 const NameAttrList func_name_attrs; member
|
| D | extract_outside_compilation_pass.h | 92 const NameAttrList& func_name_attrs, const string& new_func_name,
|
| D | extract_outside_compilation_pass.cc | 2309 const NameAttrList& func_name_attrs, const string& new_func_name, in ExtractOutsideCompilationForFunction() argument
2315 const string& func_name = func_name_attrs.name(); in ExtractOutsideCompilationForFunction()
2318 flr->Instantiate(func_name, AttrSlice(&func_name_attrs.attr()), &handle)); in ExtractOutsideCompilationForFunction()
2514 auto const& func_name_attrs = iter.second.func_name_attrs; in ExtractOutsideCompilation() local
2523 func_name_attrs, func_name_attrs.name(), host_graph_func_name, in ExtractOutsideCompilation()
|
| D | extract_outside_compilation_pass_test.cc | 241 const string &xla_cluster_name, const NameAttrList &func_name_attrs, in ExtractOutsideCompilationTest() argument
255 func_name_attrs, new_func_name, host_graph_func_name, host_compute_core, in ExtractOutsideCompilationTest()
|
| D | encapsulate_subgraphs_pass_test.cc | 526 NameAttrList func_name_attrs; in Encapsulate() local
527 func_name_attrs.set_name(func); in Encapsulate()
529 XlaClusterInfo{func, func_name_attrs, xla_computation_node, in Encapsulate()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | rearrange_function_argument.cc | 523 NameAttrList func_name_attrs; in RearrangeFunctionArguments() local
524 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "f", &func_name_attrs)); in RearrangeFunctionArguments()
526 TF_RETURN_IF_ERROR(get_function_body_fn(func_name_attrs, &fbody)); in RearrangeFunctionArguments()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 2899 string xla_func_name = iter.second.func_name_attrs.name(); in ProcessHeadTailOutsideCompilation()
2903 AttrSlice(&iter.second.func_name_attrs.attr()), fld, &xla_fbody)); in ProcessHeadTailOutsideCompilation()
2919 outside_compilation_attr_name, iter.second.func_name_attrs.name(), in ProcessHeadTailOutsideCompilation()
2924 outside_compilation_attr_name, iter.second.func_name_attrs.name(), in ProcessHeadTailOutsideCompilation()
2936 iter.second.func_name_attrs.name(), g, xla_graph, iter.second.node)); in ProcessHeadTailOutsideCompilation()
2940 TF_RETURN_IF_ERROR(RemoveUnusedXlaInput(iter.second.func_name_attrs.name(), in ProcessHeadTailOutsideCompilation()
2945 TF_RETURN_IF_ERROR(RemoveUnusedXlaOutput(iter.second.func_name_attrs.name(), in ProcessHeadTailOutsideCompilation()
|