| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| implementation_selector_test.cc | 63 auto* func_attr = cpu_def.mutable_attr(); local
64 (*func_attr)["api_implements"].set_s("times_two");
65 (*func_attr)["api_preferred_device"].set_s("CPU");
101 auto* func_attr = cpu_def.mutable_attr(); local
102 (*func_attr)["api_implements"].set_s("random_boost");
103 (*func_attr)["api_preferred_device"].set_s("CPU");
|
| function_api_info_test.cc | 59 auto* func_attr = func_def->mutable_attr(); local
61 (*func_attr)["api_implements"].set_s(api_interface_name);
63 (*func_attr)["api_preferred_device"].set_s(preferred_device);
65 (*func_attr)["forward_function_name"].set_s(forward_function_name);
67 (*func_attr)["backward_function_name"].set_s(backward_function_name);
|
| function_optimizer.cc | 112 auto* func_attr = AttrSlice(func_node).Find(kFuncAttrName); local
113 return func_attr != nullptr && func_attr->has_func() &&
114 func_attr->func().name() == func.signature().name();
123 auto* func_attr = AttrSlice(func_node).Find(kFuncAttrName); local
124 return AttrSlice(&func_attr->func().attr());
479 const AttrValue* func_attr = AttrSlice(node).Find("f"); local
480 return (func_attr != nullptr && func_attr->has_func())
481 ? ctx.function_library().Find(func_attr->func().name()
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| functionalize_control_flow.cc | 261 const string func_attr = it->second; local
263 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), func_attr, &func));
273 n->ClearAttr(func_attr);
275 n->AddAttr(func_attr, func);
|
| tf2xla_util.cc | 180 NameAttrList func_attr; local
181 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), attr_name, &func_attr));
182 const FunctionDef* fdef = lookup_fld->Find(func_attr.name());
184 return errors::Internal("Cannot find function ", func_attr.name(),
189 *fdef, AttrSlice(&func_attr.attr()), lookup_fld,
204 fld->UniqueFunctionName(absl::StrCat(func_attr.name(), "_const_"));
210 func_attr.set_name(new_func_name);
212 n->AddAttr(attr_name, func_attr);
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| functions.h | 173 const AttrSlice& func_attr() const;
192 AttrSlice func_attr,
|
| functions.cc | 311 string func_name, string description, AttrSlice func_attr,
317 func_attr_(func_attr),
383 const AttrSlice& GrapplerFunctionItem::func_attr() const { return func_attr_; } function in class:tensorflow::grappler::GrapplerFunctionItem
650 /*func_attr=*/AttrSlice(&func.attr()), std::move(inputs),
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| map_vectorization.cc | 281 auto& func_attr = (*map_node.mutable_attr())["f"]; local
282 func_attr.mutable_func()->set_name(vectorized_func.signature().name());
386 NameAttrList func_attr; local
387 func_attr.set_name(branch_func->signature().name());
388 branches->push_back(std::move(func_attr));
|
| vectorization_utils.cc | 627 AttrValue func_attr; local
628 func_attr.mutable_func()->set_name(map_defun_fn->signature().name());
629 map_defun_node_->AddAttr("f", func_attr);
|
| vectorization_utils_test.cc | 71 NameAttrList func_attr; local
72 func_attr.set_name(inner.signature().name());
77 .Attr("f", func_attr)
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| functional_ops.py | [all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_function.cc | 166 for (const auto& func_attr : fdef->signature().attr()) {
167 func_attr_names.insert(func_attr.name());
|
| /external/tensorflow/tensorflow/python/framework/ |
| ops_test.py | 457 op = test_ops.func_attr(func)
[all...] |
