Searched refs:f_attr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_and_batch_fusion_test.cc | 52 AttrValue f_attr; in TEST() local
53 SetAttrValue("f", &f_attr); in TEST()
54 map_attrs[0] = std::make_pair("f", f_attr); in TEST()
135 AttrValue f_attr; in TEST() local
136 SetAttrValue("f", &f_attr); in TEST()
137 map_attrs[0] = std::make_pair("f", f_attr); in TEST()
222 AttrValue f_attr; in TEST() local
223 SetAttrValue("f", &f_attr); in TEST()
224 map_attrs[0] = std::make_pair("f", f_attr); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | take_while_dataset_op.cc | 153 AttrValue f_attr; in AsGraphDefInternal() local
154 b->BuildAttrValue(func_, &f_attr); in AsGraphDefInternal()
162 {std::make_pair("predicate", f_attr), in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | parallel_map_dataset_op.cc | 169 AttrValue f_attr; in AsGraphDefInternal() local
170 b->BuildAttrValue(func_, &f_attr); in AsGraphDefInternal()
194 {std::make_pair("f", f_attr), in AsGraphDefInternal()
|
| D | map_dataset_op.cc | 162 AttrValue f_attr; in AsGraphDefInternal() local
163 b->BuildAttrValue(func_, &f_attr); in AsGraphDefInternal()
181 {std::make_pair("f", f_attr), in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 560 const auto* f_attr = gtl::FindOrNull(node.attr(), "f"); in Optimize() local
561 if (f_attr) differentiable_functions.insert(f_attr->func().name()); in Optimize()
|
| D | function_optimizer.cc | 2070 const auto* f_attr = gtl::FindOrNull(node->attr(), "f"); in RunFunctionOptimizerPass() local
2071 const string f_name = f_attr != nullptr ? f_attr->func().name() : ""; in RunFunctionOptimizerPass()
|