Searched refs:LowerFunctionalOpsPass (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_functional_ops.h | 33 class LowerFunctionalOpsPass : public GraphOptimizationPass {
35 LowerFunctionalOpsPass() = default;
36 LowerFunctionalOpsPass(bool keep_lowered_nodes_fetchable) in LowerFunctionalOpsPass() function
|
| D | lower_functional_ops.cc | 80 Status LowerFunctionalOpsPass::Run( in Run()
181 LowerFunctionalOpsPass);
|
| D | lower_if_op_test.cc | 59 LowerFunctionalOpsPass pass; in Rewrite()
85 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
205 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
292 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
|
| D | lower_while_op_test.cc | 53 LowerFunctionalOpsPass pass; in Rewrite()
81 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
203 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
310 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
431 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
546 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
|
| D | lower_case_op_test.cc | 61 LowerFunctionalOpsPass pass; in Rewrite()
89 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
241 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
358 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in TEST()
|
| D | lower_functional_ops_test.cc | 41 LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr;
63 LowerFunctionalOpsPass pass; in Rewrite()
|
| D | lower_function_call_op_test.cc | 66 LowerFunctionalOpsPass pass; in Rewrite()
|
| D | executor_test.cc | 633 .Attr(LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr, true) in BM_WhileLoopHelper()
651 LowerFunctionalOpsPass pass; in BM_WhileLoopHelper()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 789 LowerFunctionalOpsPass::kLowerUsingSwitchMergeAttr;
791 LowerFunctionalOpsPass::kLowerAsMultiDeviceFunctionAttr;
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 3786 n->ClearAttr(LowerFunctionalOpsPass::kLowerAsMultiDeviceFunctionAttr); in LowerOutsideCompilationFunctionalNodes()
3788 n->ClearAttr(LowerFunctionalOpsPass::kLowerAsMultiDeviceFunctionAttr); in LowerOutsideCompilationFunctionalNodes()
3789 n->AddAttr(LowerFunctionalOpsPass::kLowerAsMultiDeviceFunctionAttr, in LowerOutsideCompilationFunctionalNodes()
|