Searched refs:has_ref_vars (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 156 bool has_ref_vars) in XlaLocalLaunchBase() argument
162 has_ref_vars_(has_ref_vars) {} in XlaLocalLaunchBase()
165 OpKernelContext* ctx, const NameAttrList& function, bool has_ref_vars, in CompileToLocalExecutable() argument
197 platform_info, has_ref_vars, &tf_allocator_adapter); in CompileToLocalExecutable()
204 compile_options.alias_resource_update = !has_ref_vars && in CompileToLocalExecutable()
345 bool has_ref_vars; in HasRefVars() local
347 ctx->GetAttr(kXlaHasReferenceVarsAttr, &has_ref_vars)); in HasRefVars()
348 return has_ref_vars; in HasRefVars()
|
| D | xla_ops.h | 52 const NameAttrList& function, bool has_ref_vars);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_platform_info.cc | 133 se::Stream* stream, const XlaPlatformInfo& platform_info, bool has_ref_vars, in GenerateCompilerOptions() argument
154 !has_ref_vars && !platform_info.is_on_xla_device(); in GenerateCompilerOptions()
|
| D | xla_platform_info.h | 107 se::Stream* stream, const XlaPlatformInfo& platform_info, bool has_ref_vars,
|
| D | encapsulate_subgraphs_pass.cc | 1319 bool has_ref_vars = ref_related_nodes.contains(node); in Run() local
1320 node->AddAttr(kXlaHasReferenceVarsAttr, has_ref_vars); in Run()
1321 VLOG(3) << "Has ref vars = " << has_ref_vars in Run()
|