Home
last modified time | relevance | path

Searched defs:func_attr (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dimplementation_selector_test.cc202 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
260 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
306 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
344 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
382 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
Dfunction_api_info_test.cc59 auto* func_attr = func_def->mutable_attr(); in PopulateFunction() local
Dfunction_optimizer.cc89 auto* func_attr = AttrSlice(func_node).Find(kFuncAttr); in IsIndirectFunctionCall() local
100 auto* func_attr = AttrSlice(func_node).Find(kFuncAttr); in FunctionInstantiationAttributes() local
391 const AttrValue* func_attr = AttrSlice(node).Find("f"); in FindFunctionCall() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dperception_ops_utils_test.cc163 auto func_attr = createMaxUnpoolingAttr( in TEST_F() local
174 auto func_attr = createMaxUnpoolingAttr( in TEST_F() local
185 auto func_attr = createMaxUnpoolingAttr( in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_attr.cc99 for (const auto& func_attr : value.func().attr()) { in ConvertAttributeValue() local
Dexport_utils.cc368 if (auto func_attr = attr.dyn_cast<mlir::TF::FuncAttr>()) { in ConvertAttributes() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization.cc295 auto& func_attr = (*map_node.mutable_attr())["f"]; in AddNewMapNode() local
407 NameAttrList func_attr; in AddBranch() local
Dvectorization_utils.cc620 AttrValue func_attr; in GetResult() local
Dvectorization_utils_test.cc71 NameAttrList func_attr; in WrapFunctionWithMapDefun() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc40 string func_name, string description, AttrSlice func_attr, in GrapplerFunctionItem()
111 const AttrSlice& GrapplerFunctionItem::func_attr() const { return func_attr_; } in func_attr() function in tensorflow::grappler::GrapplerFunctionItem
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc359 const string func_attr = it->second; in Run() local
Dtf2xla_util.cc180 NameAttrList func_attr; in PropagateConstIntoFuncAttr() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.cc347 else if (auto func_attr = attr.dyn_cast<FuncAttr>()) in printAttribute() local
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc162 for (const auto& func_attr : fdef->signature().attr()) { in FillFunctionBody() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc387 FlatSymbolRefAttr func_attr = cluster_func.funcAttr(); in BuildCompileOp() local
/external/tensorflow/tensorflow/core/kernels/data/
Dhash_utils_test.cc1004 std::pair<string, FunctionDefHelper::AttrValueWrapper> func_attr = { in TEST_F() local
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc573 NameAttrList* func_attr, Node* callsite_node, in AddFunctionWithNewName()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc2988 string func_attr = iter->second; in Run() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1186 for (const auto& func_attr : value.func().attr()) { in ConvertAttributeValue() local