Home
last modified time | relevance | path

Searched defs:arg_attrs (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/ir/importexport/
Dfunctiondef_export.cc62 DictionaryAttr arg_attrs = function_interface_impl::getArgAttrDict( in GetValueName() local
105 static Status ExportArgDef(OpDef::ArgDef *arg, DictionaryAttr arg_attrs, in ExportArgDef()
217 DictionaryAttr arg_attrs = args_attr[arg_num].cast<DictionaryAttr>(); in ConvertGenericFunctionToFunctionDef() local
Dgraphdef_import.cc550 SmallVector<Attribute> arg_attrs, res_attrs, control_ret_attrs; in ConvertFunctionDef() local
/external/tensorflow/tensorflow/core/transforms/graph_to_func/
Dgraph_to_func.cc93 llvm::SmallVector<NamedAttribute> arg_attrs; in GraphToFunc() local
106 llvm::SmallVector<NamedAttribute> arg_attrs; in GraphToFunc() local
/external/tensorflow/tensorflow/core/common_runtime/
Dcontrol_flow_deps_to_chains.cc223 FunctionDef_ArgAttrs arg_attrs; in Run() local
290 FunctionDef_ArgAttrs arg_attrs; in Run() local
/external/tensorflow/tensorflow/core/transforms/utils/
Dutils.cc106 UpdateArgAttrsIfPresent(region, [&](SmallVectorImpl<Attribute> &arg_attrs) { in LoopRegionAddArgument()
123 UpdateArgAttrsIfPresent(region, [&](SmallVectorImpl<Attribute> &arg_attrs) { in LoopRegionEraseArgument()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Doutline_with_xla_framework.cc96 SmallVector<DictionaryAttr> arg_attrs; in matchAndRewrite() local
/external/tensorflow/tensorflow/core/transforms/consolidate_attrs/
Dpass.cc175 SmallVector<Attribute> arg_attrs; in reifyAndDropFunctionArgumentAttributes() local
368 ArrayAttr arg_attrs = reifyAndDropFunctionArgumentAttributes(func); in runOnOperation() local
404 SmallVector<Attribute> input_shapes, arg_attrs, res_attrs; in prepareFunctionAttributes() local
/external/tensorflow/tensorflow/core/transforms/functional_to_region/
Dimpl.cc218 ArrayAttr arg_attrs, ArrayAttr res_attrs) { in ArePreservedAttrsEmpty()
254 ArrayAttr arg_attrs = drop_args || !func.arg_attrs() in PreserveAttributes() local
/external/tensorflow/tensorflow/core/transforms/graph_compactor/
Dpass.cc106 SmallVector<Attribute> arg_attrs; in runOnOperation() local
/external/tensorflow/tensorflow/core/ir/
Dops.cc524 SmallVector<Attribute> arg_attrs; in parse() local
681 if (auto arg_attrs = args_attr[i].dyn_cast<DictionaryAttr>()) in print() local
767 DictionaryAttr arg_attrs = args_attr[arg_num].dyn_cast<DictionaryAttr>(); in getAsmBlockArgumentNames() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc230 SmallVectorImpl<Value>* inputs, NamedAttrList* arg_attrs, in CollectInputsAndAttributes()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc429 FunctionDef::ArgAttrs arg_attrs; in GraphToFunctionDef() local
Dfunction_test.cc1580 FunctionDef::ArgAttrs arg_attrs; in TEST() local
1641 FunctionDef::ArgAttrs arg_attrs; in TEST() local
Dfunction.cc177 const FunctionDef::ArgAttrs* arg_attrs, in BuildInputArgIndex()
767 const FunctionDef::ArgAttrs* arg_attrs = in InstantiateFunction() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtfg-to-tfe.cc191 llvm::SmallVector<DictionaryAttr> arg_attrs; in matchAndRewrite() local
/external/tensorflow/tensorflow/core/transforms/region_to_functional/
Dimpl.cc509 SmallVector<Attribute> arg_attrs, res_attrs; in BuildAttributes() local
/external/tensorflow/tensorflow/c/
Dc_api_function_test.cc1312 auto arg_attrs = func_->fdef.arg_attr().find(0); in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dtf_to_tfrt.cc693 llvm::SmallVector<mlir::DictionaryAttr, 4> arg_attrs; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc1746 llvm::SmallVector<mlir::DictionaryAttr, 4> arg_attrs, res_attrs; in BuildSignaturedef() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1609 llvm::SmallVector<mlir::NamedAttrList, 4> arg_attrs; in ConvertFunctionArgAndRets() local