- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for has_ref_vars_ (0.42 sec)
-
tensorflow/compiler/jit/kernels/xla_ops.cc
const NameAttrList& function, bool has_ref_vars) : AsyncOpKernel(ctx), constants_(constants), resources_(resources), function_(function), platform_info_(XlaPlatformInfoFromDevice(ctx->device())), has_ref_vars_(has_ref_vars) {} void XlaLocalLaunchBase::ComputeAsync(OpKernelContext* ctx, DoneCallback done) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util.cc
bool has_ref_vars, bool may_alias_resource_update) { XlaCompiler::CompileOptions compile_options; compile_options.is_entry_computation = true; // Optimization: where possible, have the computation return a naked array // rather than a one-element tuple. compile_options.always_return_tuple = false; compile_options.alias_resource_update = !has_ref_vars && may_alias_resource_update;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
**xla_device_compiler, *ctx->function_library(), ctx->device(), ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr, platform_info_, /*has_ref_vars=*/true); // No detailed logging from on demand op. options.detailed_logging = false; XlaCompiler::CompileOptions compile_options = GetCompileOptions(); return (*xla_device_compiler)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
EncapsulateSubgraphsPass pass; TF_ASSERT_OK(pass.Run(options)); for (const Node* node : graph->nodes()) { bool has_ref_var; TF_ASSERT_OK( GetNodeAttr(node->attrs(), kXlaHasReferenceVarsAttr, &has_ref_var)); EXPECT_TRUE(node->IsSink() || node->IsSource() || has_ref_var) << "All nodes apart from source and sink can access reference variable"; } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
GetNodesRelatedToRefVariables(**options.graph, flr)); for (Node* node : (*options.graph)->nodes()) { bool has_ref_vars = ref_related_nodes.contains(node); node->AddAttr(kXlaHasReferenceVarsAttr, has_ref_vars); VLOG(3) << "Has ref vars = " << has_ref_vars << ", node: " << node->def().DebugString(); } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
} else { options = GenerateCompilerOptions(*xla_device_compiler, *flr, dev, stream, platform_info, /*has_ref_vars=*/false); } return CompileAndBuildHLOString(stage, options, xla_device_compiler->client(), function, args); } absl::StatusOr<std::string> GetCompilerIr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0)