Searched refs:n_attrs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | parallel_concat_optimizer.cc | 52 AttrSlice n_attrs = n->attrs(); in Run() local
53 auto base_make_node = [n, &n_attrs](const string& op, in Run()
59 if (GetNodeAttr(n_attrs, "_class", &colo).ok()) { in Run()
69 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in Run()
71 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "shape", &shape)); in Run()
|
| D | accumulate_n_optimizer.cc | 76 AttrSlice n_attrs = n->attrs(); in rewriteNode() local
77 auto base_make_node = [n, &n_attrs](const string& op, const string& name) { in rewriteNode()
84 if (GetNodeAttr(n_attrs, kColocationAttrName, &colo).ok()) { in rewriteNode()
95 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in rewriteNode()
97 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "shape", &shape)); in rewriteNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 84 AttrSlice n_attrs = AttrSlice(*n); in CheckTypesAndGetShapes() local
86 LOG_WARNING_AND_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in CheckTypesAndGetShapes()
224 AttrSlice n_attrs = AttrSlice(*nd.from_node_def); in CheckExistingScopedAllocator() local
226 Status ss = GetNodeAttr(n_attrs, "sa_id", &sa_id); in CheckExistingScopedAllocator()
|
| /external/perf_data_converter/src/quipper/ |
| D | test_perf_data.cc | 80 size_t n_attrs = header_.attrs.size / header_.attr_size; in WithCustomPerfEventAttrSize() local
84 return WithAttrCount(n_attrs); in WithCustomPerfEventAttrSize()
|