Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A 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
H A 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
394 const AttrValue* func_attr = AttrSlice(node).Find("f"); in FindFunctionCall() local
H A Dfunction_api_info_test.cc59 auto* func_attr = func_def->mutable_attr(); in PopulateFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state_test.cc357 tensorflow::NameAttrList func_attr; in TEST_F() local
424 tensorflow::NameAttrList func_attr; in TEST_F() local
477 tensorflow::NameAttrList func_attr; in TEST_F() local
559 tensorflow::NameAttrList func_attr; in TEST_F() local
600 tensorflow::NameAttrList func_attr; in TEST_F() local
667 tensorflow::NameAttrList func_attr; in TEST_F() local
933 tensorflow::NameAttrList func_attr; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dperception_ops_utils_test.cc165 auto func_attr = createMaxUnpoolingAttr( in TEST_F() local
176 auto func_attr = createMaxUnpoolingAttr( in TEST_F() local
187 auto func_attr = createMaxUnpoolingAttr( in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dconvert_attributes.cc207 if (auto func_attr = attr.dyn_cast<FuncAttr>()) { in ConvertAttribute() local
350 for (const auto& func_attr : value.list().func()) { in ConvertNonFuncAttributeValue() local
380 for (const auto& func_attr : value.func().attr()) { in ConvertAttributeValue() local
H A Dfunctiondef_export.cc173 OpDef_AttrDef *func_attr = signature->add_attr(); in ConvertGenericFunctionToFunctionDef() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dattr_util.cc172 auto func_attr = list_attr.GetAttributeOfType<tfrt::FuncAttr>(i); in FillAttrValueMapUsingAggregate() local
577 Status SetUpScalarFunctionAttr(tfrt::StringAttr func_attr, in SetUpScalarFunctionAttr()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/types/
H A Ddialect_test.cc49 auto func_attr = test_op.getAttr("func").dyn_cast<tf_type::FuncAttr>(); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dconvert_attr.cc102 for (const auto& func_attr : value.func().attr()) { in ConvertAttributeValue() local
H A Dexport_utils.cc396 if (auto func_attr = attr.dyn_cast<mlir::TF::FuncAttr>()) { in ConvertAttributes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dmark_input_output_aliases.cc135 FlatSymbolRefAttr func_attr = cluster_func.funcAttr(); in runOnOperation() local
H A Dtpu_rewrite_pass.cc363 FlatSymbolRefAttr func_attr = cluster_func.funcAttr(); in BuildCompileOp() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc766 AttrValue func_attr; in PrepareGraphForMlir() local
1131 NameAttrList func_attr; in InsertFunctionForTPUEmbeddingCheckpoint() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfunctionalize_control_flow.cc358 const string func_attr = it->second; in Run() local
H A Dtf2xla_util.cc235 NameAttrList func_attr; in PropagateConstIntoFuncAttr() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dcorert_converter.cc108 auto func_attr = attr_value.dyn_cast<mlir::FlatSymbolRefAttr>(); in CreateOpFuncAttrs() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dgraph_to_functiondef.cc162 for (const auto& func_attr : fdef->signature().attr()) { in FillFunctionBody() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc504 auto func_attr = list_attr.GetAttributeOfType<FuncAttr>(i); in PrepareAttributes() local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dhash_utils_test.cc1007 std::pair<string, FunctionDefHelper::AttrValueWrapper> func_attr = { in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/
H A Dops.cc727 auto func_attr = op->getAttrOfType<FuncAttr>("f"); in getCalledFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dextract_outside_compilation_pass.cc569 NameAttrList* func_attr, Node* callsite_node, in AddFunctionWithNewName()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc2936 string func_attr = iter->second; in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dimport_model.cc1339 for (const auto& func_attr : value.func().attr()) { in ConvertAttributeValue() local