Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for has_ref_vars_ (0.42 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
Back to top