Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for GetNodeAttr (0.45 sec)

  1. tensorflow/cc/gradients/math_grad.cc

        a = ConjugateHelper(scope, a);
        b = ConjugateHelper(scope, b);
      }
      auto product = op.output(0);
    
      bool ta;
      bool tb;
      TF_RETURN_IF_ERROR(GetNodeAttr(product.node()->attrs(), attr_adj_x, &ta));
      TF_RETURN_IF_ERROR(GetNodeAttr(product.node()->attrs(), attr_adj_y, &tb));
    
      if (!ta && !tb) {
        return MatMulGradHelper(scope, is_batch, grad_inputs[0], false, b, true,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 18:20:20 UTC 2023
    - 50.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

    static Status GetArgTypes(const Graph& graph, DataTypeVector* types) {
      for (Node* n : graph.op_nodes()) {
        if (n->type_string() == kArgOp) {
          int index;
          TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index));
          const int num_types = types->size();
          if (index < 0 || index >= num_types) {
            return errors::InvalidArgument("Invalid argument number");
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/deadness_analysis.cc

        TF_RET_CHECK(!must_be_true || is_boolean_tensor);
    
        if (node->type_string() == "Const" && must_be_true) {
          const TensorProto* proto = nullptr;
          TF_RETURN_IF_ERROR(GetNodeAttr(node->def(), "value", &proto));
    
          Tensor tensor(proto->dtype());
          TF_RET_CHECK(tensor.FromProto(*proto));
    
          *predicate = tensor.scalar<bool>()() ? MakeTrue() : MakeFalse();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 60.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        Node* node) {
      if (debug_options_.ignore_xla_compile_attr) {
        return false;
      }
    
      // If there is a _XlaCompile annotation, use its value.
      bool compile = false;
      Status status = GetNodeAttr(node->attrs(), kXlaCompileAttr, &compile);
      if (status.ok()) {
        if (!compile) {
          VLOG(2) << "Rejecting " << node->name() << ": kXlaCompileAttr("
                  << kXlaCompileAttr << ") is false.";
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top