| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | implementation_selector_test.cc | 202 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
203 (*func_attr)["api_implements"].set_s("times_two"); in TEST_F()
204 (*func_attr)["api_preferred_device"].set_s("CPU"); in TEST_F()
260 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
261 (*func_attr)["api_implements"].set_s("times_two"); in TEST_F()
306 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
307 (*func_attr)["api_implements"].set_s("times_two"); in TEST_F()
308 (*func_attr)["api_preferred_device"].set_s("CPU"); in TEST_F()
344 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
345 (*func_attr)["api_implements"].set_s("times_two"); in TEST_F()
[all …]
|
| 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()
391 const AttrValue* func_attr = AttrSlice(node).Find("f"); in FindFunctionCall() local
392 return (func_attr != nullptr && func_attr->has_func()) in FindFunctionCall()
393 ? ctx.function_library().Find(func_attr->func().name()) in FindFunctionCall()
|
| /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/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/llvm-project/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 {
|
| D | tailcc-disable-tail-calls.ll | 8 ; NO-OPTION-LABEL: {{\_?}}func_attr
11 ; DISABLE-FALSE-LABEL: {{\_?}}func_attr
14 ; DISABLE-TRUE-LABEL: {{\_?}}func_attr
17 define tailcc i32 @func_attr(i32 %a) #0 {
|
| /external/llvm-project/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/mlir/lite/utils/ |
| D | perception_ops_utils_test.cc | 163 auto func_attr = createMaxUnpoolingAttr( in TEST_F() local
168 func_attr); in TEST_F()
174 auto func_attr = createMaxUnpoolingAttr( in TEST_F() local
179 func_attr); in TEST_F()
185 auto func_attr = createMaxUnpoolingAttr( in TEST_F() local
190 func_attr); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_attr.cc | 99 for (const auto& func_attr : value.func().attr()) { in ConvertAttributeValue() local
101 ConvertAttributeValue(func_attr.second, builder)); in ConvertAttributeValue()
102 attrs.push_back(builder->getNamedAttr(func_attr.first, attr)); in ConvertAttributeValue()
|
| D | export_utils.cc | 368 if (auto func_attr = attr.dyn_cast<mlir::TF::FuncAttr>()) { in ConvertAttributes() local
369 TF_RETURN_IF_ERROR(ConvertAttribute(func_attr, remove_ref_type, &value)); in ConvertAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | func-attr.mlir | 3 // CHECK-LABEL: func @func_attr
5 func @func_attr() attributes {tf._implements = #tf.func<@symbol_a, {attr0 = 1 : i32, attr1 = "rando…
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| 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()
203 replace_fdef, lookup_fld->GetStackTraces(func_attr.name()))); in PropagateConstIntoFuncAttr()
206 func_attr.set_name(new_func_name); in PropagateConstIntoFuncAttr()
208 n->AddAttr(attr_name, func_attr); in PropagateConstIntoFuncAttr()
211 replace_fdef, lookup_fld->GetStackTraces(func_attr.name()))); in PropagateConstIntoFuncAttr()
|
| D | functionalize_control_flow.cc | 359 const string func_attr = it->second; in Run() local
361 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), func_attr, &func)); in Run()
371 n->ClearAttr(func_attr); in Run()
373 n->AddAttr(func_attr, func); in Run()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.h | 78 const AttrSlice& func_attr() const;
98 AttrSlice func_attr,
|
| D | functions.cc | 40 string func_name, string description, AttrSlice func_attr, in GrapplerFunctionItem() argument
47 func_attr_(func_attr), in GrapplerFunctionItem()
111 const AttrSlice& GrapplerFunctionItem::func_attr() const { return func_attr_; } in func_attr() function in tensorflow::grappler::GrapplerFunctionItem
578 for (const auto& attr : item.func_attr()) { in MakeFunctionDef()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization.cc | 295 auto& func_attr = (*map_node.mutable_attr())["f"]; in AddNewMapNode() local
296 func_attr.mutable_func()->set_name(vectorized_func.signature().name()); in AddNewMapNode()
407 NameAttrList func_attr; in AddBranch() local
408 func_attr.set_name(branch_func->signature().name()); in AddBranch()
409 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/ir/ |
| D | tf_ops.cc | 347 else if (auto func_attr = attr.dyn_cast<FuncAttr>()) in printAttribute() local
348 PrintFuncAttr(func_attr, os); in printAttribute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.cc | 162 for (const auto& func_attr : fdef->signature().attr()) { in FillFunctionBody() local
163 func_attr_names.insert(func_attr.name()); in FillFunctionBody()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | functional_ops.py | 1195 func_attr = attr_value_pb2.AttrValue(
1215 "f": func_attr,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 387 FlatSymbolRefAttr func_attr = cluster_func.funcAttr(); in BuildCompileOp() local
389 func_attr.getValue()); in BuildCompileOp()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | hash_utils_test.cc | 1004 std::pair<string, FunctionDefHelper::AttrValueWrapper> func_attr = { in TEST_F() local
1015 {{"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 | 1158 TF_ASSIGN_OR_RETURN(auto func_attr, in ConvertFunctionCallAttribute()
1160 attributes->push_back(builder_.getNamedAttr(base_name, func_attr)); in ConvertFunctionCallAttribute()
1186 for (const auto& func_attr : value.func().attr()) { in ConvertAttributeValue() local
1188 auto attr, ImporterBase::ConvertAttributeValue(func_attr.second)); in ConvertAttributeValue()
1189 attrs.push_back(builder_.getNamedAttr(func_attr.first, attr)); in ConvertAttributeValue()
|