Searched refs:kXlaHasReferenceVarsAttr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass_test.cc | 142 call->AddAttr(kXlaHasReferenceVarsAttr, false); in TEST_F()
145 write_op->AddAttr(kXlaHasReferenceVarsAttr, false); in TEST_F()
186 call->AddAttr(kXlaHasReferenceVarsAttr, false); in TEST_F()
189 write_op->AddAttr(kXlaHasReferenceVarsAttr, false); in TEST_F()
223 call->AddAttr(kXlaHasReferenceVarsAttr, false); in TEST_F()
226 write_op->AddAttr(kXlaHasReferenceVarsAttr, false); in TEST_F()
249 call->AddAttr(kXlaHasReferenceVarsAttr, false); in TEST_F()
290 call->AddAttr(kXlaHasReferenceVarsAttr, false); in TEST_F()
|
| D | encapsulate_subgraphs_pass.h | 104 extern const char* const kXlaHasReferenceVarsAttr;
|
| D | build_xla_ops_pass.cc | 504 GetNodeAttr(n->attrs(), kXlaHasReferenceVarsAttr, &has_ref_attr)); in ReplaceNodeWithXlaCompileAndXlaRun()
505 xla_compile.operation.node()->AddAttr(kXlaHasReferenceVarsAttr, has_ref_attr); in ReplaceNodeWithXlaCompileAndXlaRun()
|
| D | encapsulate_subgraphs_pass.cc | 65 const char* const kXlaHasReferenceVarsAttr = "_XlaHasReferenceVars"; variable
1320 node->AddAttr(kXlaHasReferenceVarsAttr, has_ref_vars); in Run()
|
| D | encapsulate_subgraphs_pass_test.cc | 2677 GetNodeAttr(node->attrs(), kXlaHasReferenceVarsAttr, &has_ref_var)); in TEST()
2707 GetNodeAttr(node->attrs(), kXlaHasReferenceVarsAttr, &has_ref_var)); in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 347 ctx->GetAttr(kXlaHasReferenceVarsAttr, &has_ref_vars)); in HasRefVars()
|