| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | implementation_selector_test.cc | 64 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
65 (*func_attr)["api_implements"].set_s("times_two"); in TEST_F()
66 (*func_attr)["api_preferred_device"].set_s("CPU"); in TEST_F()
102 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
103 (*func_attr)["api_implements"].set_s("times_two"); in TEST_F()
104 (*func_attr)["api_preferred_device"].set_s("CPU"); in TEST_F()
140 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
141 (*func_attr)["api_implements"].set_s("random_boost"); in TEST_F()
142 (*func_attr)["api_preferred_device"].set_s("CPU"); in TEST_F()
|
| D | function_api_info_test.cc | 59 auto* func_attr = func_def->mutable_attr(); in PopulateFunction() local
61 (*func_attr)["api_implements"].set_s(api_interface_name); in PopulateFunction()
63 (*func_attr)["api_preferred_device"].set_s(preferred_device); in PopulateFunction()
65 (*func_attr)["forward_function_name"].set_s(forward_function_name); in PopulateFunction()
67 (*func_attr)["backward_function_name"].set_s(backward_function_name); in PopulateFunction()
|
| D | function_optimizer.cc | 89 auto* func_attr = AttrSlice(func_node).Find(kFuncAttr); in IsIndirectFunctionCall() local
90 return func_attr != nullptr && func_attr->has_func() && in IsIndirectFunctionCall()
91 func_attr->func().name() == func.signature().name(); in IsIndirectFunctionCall()
100 auto* func_attr = AttrSlice(func_node).Find(kFuncAttr); in FunctionInstantiationAttributes() local
101 return AttrSlice(&func_attr->func().attr()); in FunctionInstantiationAttributes()
384 const AttrValue* func_attr = AttrSlice(node).Find("f"); in FindFunctionCall() local
385 return (func_attr != nullptr && func_attr->has_func()) in FindFunctionCall()
386 ? ctx.function_library().Find(func_attr->func().name()) in FindFunctionCall()
|
| /external/llvm/test/CodeGen/X86/ |
| D | disable-tail-calls.ll | 8 ; NO-OPTION-LABEL: {{\_?}}func_attr
11 ; DISABLE-FALSE-LABEL: {{\_?}}func_attr
14 ; DISABLE-TRUE-LABEL: {{\_?}}func_attr
17 define i32 @func_attr(i32 %a) #0 {
|
| /external/swiftshader/third_party/llvm-7.0/llvm/test/CodeGen/X86/ |
| D | disable-tail-calls.ll | 8 ; NO-OPTION-LABEL: {{\_?}}func_attr
11 ; DISABLE-FALSE-LABEL: {{\_?}}func_attr
14 ; DISABLE-TRUE-LABEL: {{\_?}}func_attr
17 define i32 @func_attr(i32 %a) #0 {
|
| /external/swiftshader/third_party/llvm-7.0/llvm/test/CodeGen/ARM/ |
| D | disable-tail-calls.ll | 8 ; NO-OPTION-LABEL: {{\_?}}func_attr
11 ; DISABLE-FALSE-LABEL: {{\_?}}func_attr
14 ; DISABLE-TRUE-LABEL: {{\_?}}func_attr
17 define i32 @func_attr(i32 %a) #0 {
|
| /external/llvm/test/CodeGen/ARM/ |
| D | disable-tail-calls.ll | 8 ; NO-OPTION-LABEL: {{\_?}}func_attr
11 ; DISABLE-FALSE-LABEL: {{\_?}}func_attr
14 ; DISABLE-TRUE-LABEL: {{\_?}}func_attr
17 define i32 @func_attr(i32 %a) #0 {
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 251 const string func_attr = it->second; in Run() local
253 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), func_attr, &func)); in Run()
263 n->ClearAttr(func_attr); in Run()
265 n->AddAttr(func_attr, func); in Run()
|
| D | tf2xla_util.cc | 180 NameAttrList func_attr; in PropagateConstIntoFuncAttr() local
181 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), attr_name, &func_attr)); in PropagateConstIntoFuncAttr()
182 const FunctionDef* fdef = lookup_fld->Find(func_attr.name()); in PropagateConstIntoFuncAttr()
184 return errors::Internal("Cannot find function ", func_attr.name(), in PropagateConstIntoFuncAttr()
189 *fdef, AttrSlice(&func_attr.attr()), lookup_fld, &fbody)); in PropagateConstIntoFuncAttr()
199 fld->UniqueFunctionName(absl::StrCat(func_attr.name(), "_const_")); in PropagateConstIntoFuncAttr()
205 func_attr.set_name(new_func_name); in PropagateConstIntoFuncAttr()
207 n->AddAttr(attr_name, func_attr); in PropagateConstIntoFuncAttr()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.h | 78 const AttrSlice& func_attr() const;
97 AttrSlice func_attr,
|
| D | functions.cc | 40 string func_name, string description, AttrSlice func_attr, in GrapplerFunctionItem() argument
46 func_attr_(func_attr), in GrapplerFunctionItem()
109 const AttrSlice& GrapplerFunctionItem::func_attr() const { return func_attr_; } in func_attr() function in tensorflow::grappler::GrapplerFunctionItem
563 for (const auto& attr : item.func_attr()) { in MakeFunctionDef()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization.cc | 282 auto& func_attr = (*map_node.mutable_attr())["f"]; in AddNewMapNode() local
283 func_attr.mutable_func()->set_name(vectorized_func.signature().name()); in AddNewMapNode()
387 NameAttrList func_attr; in AddBranch() local
388 func_attr.set_name(branch_func->signature().name()); in AddBranch()
389 branches->push_back(std::move(func_attr)); in AddBranch()
|
| D | vectorization_utils.cc | 620 AttrValue func_attr; in GetResult() local
621 func_attr.mutable_func()->set_name(map_defun_fn->signature().name()); in GetResult()
622 map_defun_node_->AddAttr("f", func_attr); in GetResult()
|
| D | vectorization_utils_test.cc | 71 NameAttrList func_attr; in WrapFunctionWithMapDefun() local
72 func_attr.set_name(inner.signature().name()); in WrapFunctionWithMapDefun()
77 .Attr("f", func_attr) in WrapFunctionWithMapDefun()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 300 FlatSymbolRefAttr func_attr = in BuildCompileOp() local
302 if (!func_attr) { in BuildCompileOp()
307 func_attr.getValue()); in BuildCompileOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.cc | 345 auto func_attr = attr.cast<mlir::FlatSymbolRefAttr>(); in ConvertAttributes() local
346 value.mutable_func()->set_name(std::string(func_attr.getValue())); in ConvertAttributes()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.cc | 161 for (const auto& func_attr : fdef->signature().attr()) { in FillFunctionBody() local
162 func_attr_names.insert(func_attr.name()); in FillFunctionBody()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | functional_ops.py | 1153 func_attr = attr_value_pb2.AttrValue(
1173 "f": func_attr,
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils_test.cc | 926 std::pair<string, FunctionDefHelper::AttrValueWrapper> func_attr = { in TEST_F() local
937 {{"for"}, "For", {"j", "j", "j"}, {func_attr, {"T", DT_FLOAT}}, {"ret"}}, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 946 TF_ASSIGN_OR_RETURN(auto func_attr, in ConvertFunctionCallAttribute()
948 attributes->push_back(builder_.getNamedAttr(base_name, func_attr)); in ConvertFunctionCallAttribute()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops_test.py | 760 op = test_ops.func_attr(func)
|