| /external/llvm-project/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/ |
| D | saved_model.pbtxt | 1763 arg_attr {
1783 arg_attr {
1797 arg_attr {
1811 arg_attr {
1825 arg_attr {
1839 arg_attr {
1853 arg_attr {
1867 arg_attr {
1881 arg_attr {
2070 arg_attr {
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 41 std::vector<const FunctionDef::ArgAttrs*> arg_attr, in GrapplerFunctionItem() argument
48 arg_attr_(std::move(arg_attr)), in GrapplerFunctionItem()
114 GrapplerFunctionItem::arg_attr() const { in arg_attr() function in tensorflow::grappler::GrapplerFunctionItem
288 std::vector<const FunctionDef::ArgAttrs*> arg_attr(inputs.size(), nullptr); in MakeGrapplerFunctionItem() local
289 for (const auto& attr : func.arg_attr()) { in MakeGrapplerFunctionItem()
290 arg_attr.at(attr.first) = &attr.second; in MakeGrapplerFunctionItem()
296 /*func_attr=*/AttrSlice(&func.attr()), std::move(arg_attr), in MakeGrapplerFunctionItem()
585 for (int i = 0, end = item.arg_attr().size(); i < end; ++i) { in MakeFunctionDef()
586 const auto* attr = item.arg_attr().at(i); in MakeFunctionDef()
|
| D | functions.h | 79 const std::vector<const FunctionDef::ArgAttrs*>& arg_attr() const;
99 std::vector<const FunctionDef::ArgAttrs*> arg_attr,
|
| D | functions_test.cc | 529 FunctionDef::ArgAttrs arg_attr; in TEST_F() local
530 (*arg_attr.mutable_attr())[arg_attr_key_and_value.first].set_s( in TEST_F()
532 (*func.mutable_arg_attr())[arg_index] = arg_attr; in TEST_F()
552 EXPECT_EQ(specialized.arg_attr().size(), 1); in TEST_F()
553 EXPECT_EQ(specialized.arg_attr().at(arg_index).attr().size(), 1); in TEST_F()
554 EXPECT_EQ(specialized.arg_attr() in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | function-resource-args.mlir | 19 // test_func_name0 are propagated to the function's arg_attr and
41 // CHECK: arg_attr
47 // CHECK: arg_attr
|
| /external/llvm-project/llvm/lib/Analysis/models/inliner/ |
| D | saved_model.pbtxt | 1633 arg_attr {
18172 arg_attr {
18195 arg_attr {
18218 arg_attr {
18241 arg_attr {
18264 arg_attr {
18287 arg_attr {
18310 arg_attr {
18333 arg_attr {
18356 arg_attr {
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | case_op.pbtxt | 169 arg_attr {
239 arg_attr {
|
| D | graph-as-function.pbtxt | 286 arg_attr {
297 arg_attr {
|
| D | tf-data-pipeline.pbtxt | 239 arg_attr {
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 205 for (const auto& arg_attr : arg_attrs->attr()) { in BuildInputArgIndex() local
206 AddAttr(arg_attr.first, arg_attr.second, gnode->mutable_attr()); in BuildInputArgIndex()
734 auto it = fdef.arg_attr().find(i); in InstantiateFunction()
736 it != fdef.arg_attr().end() ? &it->second : nullptr; in InstantiateFunction()
1002 auto iter = function_def.arg_attr().find(arg_index); in GetFunctionResourceInputDevice()
1003 if (iter != function_def.arg_attr().end()) { in GetFunctionResourceInputDevice()
1004 auto arg_attr = iter->second.attr().find("_composite_device"); in GetFunctionResourceInputDevice() local
1005 if (arg_attr != iter->second.attr().end()) { in GetFunctionResourceInputDevice()
1006 composite_device = arg_attr->second.s(); in GetFunctionResourceInputDevice()
|
| D | function.proto | 40 map<uint32, ArgAttrs> arg_attr = 7; field
|
| /external/tensorflow/tensorflow/core/data/service/testdata/ |
| D | map_graph_def.pbtxt | 200 arg_attr {
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 639 auto iter = fdef->arg_attr().find(num_non_captured_inputs + i); in Instantiate()
640 if (iter != fdef->arg_attr().end()) { in Instantiate()
641 auto arg_attr = iter->second.attr().find("_composite_device"); in Instantiate() local
642 if (arg_attr != iter->second.attr().end()) { in Instantiate()
643 composite_device = arg_attr->second.s(); in Instantiate()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | hoist_random_uniform.cc | 147 if (function->arg_attr().contains(i - 1)) { in InsertSeedArgument()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen.py | 986 arg_attr = ''
996 arg_attr += '{{tfr.name="{}"'.format(attr_def.name)
999 arg_attr += ',tfr.default={}'.format(default_val)
1000 arg_attr += '}'
1003 arg_str = '{}: {}{}'.format(arg_name, arg_type, arg_attr)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph_test.py | 229 fdef.arg_attr[0].attr["_test_attr"].s = "value".encode("ascii")
|
| D | function_def_to_graph.py | 182 arg_attrs = fdef.arg_attr[i].attr
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1557 for (const auto& arg_attr : dict_attrs) { in GetStringsFromDictionaryAttr() local
1558 if (!arg_attr) continue; in GetStringsFromDictionaryAttr()
1560 auto attrs = arg_attr.getValue(); in GetStringsFromDictionaryAttr()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function_test.cc | 1312 auto arg_attrs = func_->fdef.arg_attr().find(0); in TEST_F()
1313 ASSERT_NE(arg_attrs, func_->fdef.arg_attr().end()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 107 *(fn_stub->mutable_arg_attr()) = fn.arg_attr(); in GetFunctionDefLibraryStub()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_test.py | 241 arg_attrs = read_var.get_concrete_function().function_def.arg_attr
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | opens2s_gnmt_mixed_precision.pbtxt.gz | |