Searched refs:n_attrs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | accumulate_n_optimizer.cc | 108 AttrSlice n_attrs = n->attrs(); in RewriteIntoTempVariable() local
109 auto base_make_node = [n, &n_attrs](const string& op, const string& name) { in RewriteIntoTempVariable()
115 const string& colo = GetNodeAttrString(n_attrs, kColocationAttrName); in RewriteIntoTempVariable()
127 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in RewriteIntoTempVariable()
129 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "shape", &shape)); in RewriteIntoTempVariable()
235 AttrSlice n_attrs = n->attrs(); in RewriteIntoAddN() local
237 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in RewriteIntoAddN()
239 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "N", &num_inputs)); in RewriteIntoAddN()
264 const string& colo = GetNodeAttrString(n_attrs, kColocationAttrName); in RewriteIntoAddN()
|
| D | parallel_concat_optimizer.cc | 48 AttrSlice n_attrs = n->attrs(); in Run() local
49 auto base_make_node = [n, &n_attrs](const string& op, in Run()
54 const string& colo = GetNodeAttrString(n_attrs, "_class"); in Run()
65 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in Run()
67 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "shape", &shape)); in Run()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 99 AttrSlice n_attrs = AttrSlice(*n); in CheckTypesAndGetShapes() local
102 LOG_WARNING_AND_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in CheckTypesAndGetShapes()
388 AttrSlice n_attrs = AttrSlice(*nd.from_node_def); in CheckExistingScopedAllocator() local
390 Status ss = GetNodeAttr(n_attrs, kScopedAllocatorAttrName, &scope_ids); in CheckExistingScopedAllocator()
|