Searched refs:f_attr (Results 1 – 5 of 5) 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()
134 AttrValue f_attr; in TEST() local
135 SetAttrValue("f", &f_attr); in TEST()
136 map_attrs[0] = std::make_pair("f", f_attr); in TEST()
220 AttrValue f_attr; in TEST() local
221 SetAttrValue("f", &f_attr); in TEST()
222 map_attrs[0] = std::make_pair("f", f_attr); in TEST()
305 AttrValue f_attr; in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | take_while_dataset_op.cc | 107 AttrValue f_attr; in AsGraphDefInternal() local
108 b->BuildAttrValue(captured_func_->func(), &f_attr); in AsGraphDefInternal()
116 {std::make_pair("predicate", f_attr), in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_dataset_op.cc | 106 AttrValue f_attr; in AsGraphDefInternal() local
107 b->BuildAttrValue(captured_func_->func(), &f_attr); in AsGraphDefInternal()
125 {std::make_pair(kFunc, f_attr), in AsGraphDefInternal()
|
| D | parallel_map_dataset_op.cc | 159 AttrValue f_attr; in AsGraphDefInternal() local
160 b->BuildAttrValue(captured_func_->func(), &f_attr); in AsGraphDefInternal()
161 attrs.emplace_back(kFunc, f_attr); in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 728 const auto* f_attr = gtl::FindOrNull(node.attr(), "f"); in OptimizeConsumeItem() local
729 if (f_attr) differentiable_functions.insert(f_attr->func().name()); in OptimizeConsumeItem()
|