Searched refs:kLowerAsMultiDeviceFunctionAttr (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_function_call_op.cc | 31 constexpr const char* const kLowerAsMultiDeviceFunctionAttr = variable
32 LowerFunctionalOpsConstants::kLowerAsMultiDeviceFunctionAttr;
39 TryGetNodeAttr(n->attrs(), kLowerAsMultiDeviceFunctionAttr, &match); in LowerAsMultiDeviceFunction()
|
| D | lower_functional_ops.h | 43 static constexpr const char* const kLowerAsMultiDeviceFunctionAttr = variable
44 LowerFunctionalOpsConstants::kLowerAsMultiDeviceFunctionAttr;
|
| D | lower_functional_ops.cc | 35 constexpr const char* const kLowerAsMultiDeviceFunctionAttr = variable
36 LowerFunctionalOpsConstants::kLowerAsMultiDeviceFunctionAttr;
60 return CheckBoolAttr(n, kLowerAsMultiDeviceFunctionAttr); in LowerAsMultiDeviceFunctionIsOn()
|
| D | lower_if_op.cc | 28 constexpr const char* const kLowerAsMultiDeviceFunctionAttr = variable
29 LowerFunctionalOpsConstants::kLowerAsMultiDeviceFunctionAttr;
128 then_call_builder_.Attr(kLowerAsMultiDeviceFunctionAttr, true); in CondBuilder()
133 else_call_builder_.Attr(kLowerAsMultiDeviceFunctionAttr, true); in CondBuilder()
|
| D | lower_case_op.cc | 30 constexpr const char* const kLowerAsMultiDeviceFunctionAttr = variable
31 LowerFunctionalOpsConstants::kLowerAsMultiDeviceFunctionAttr;
113 branch_call_builders_[b].Attr(kLowerAsMultiDeviceFunctionAttr, true); in CaseBuilder()
|
| D | lower_while_op.cc | 30 constexpr const char* const kLowerAsMultiDeviceFunctionAttr = variable
31 LowerFunctionalOpsConstants::kLowerAsMultiDeviceFunctionAttr;
192 cond_call_builder_.Attr(kLowerAsMultiDeviceFunctionAttr, true); in LowerWhileHelper()
196 body_call_builder_.Attr(kLowerAsMultiDeviceFunctionAttr, true); in LowerWhileHelper()
|
| D | inline_function_utils.h | 237 static constexpr const char* const kLowerAsMultiDeviceFunctionAttr = member
|
| D | inline_function_utils.cc | 50 LowerFunctionalOpsConstants::kLowerAsMultiDeviceFunctionAttr; member in tensorflow::LowerFunctionalOpsConstants
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 790 constexpr const char* const kLowerAsMultiDeviceFunctionAttr = variable
791 LowerFunctionalOpsPass::kLowerAsMultiDeviceFunctionAttr;
814 return CheckBoolAttr(n, kLowerAsMultiDeviceFunctionAttr); in LowerAsMultiDeviceFunctionIsOn()
|
| /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()
|