Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for AddNodeAttr (0.12 sec)

  1. tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc

        def.set_device(launch->requested_device());
        AddNodeAttr("Tconstants", DataTypeVector{}, &def);
        AddNodeAttr("Targs", arg_types, &def);
        AddNodeAttr("Nresources", num_variables, &def);
        AddNodeAttr("Tresults", output_types, &def);
        NameAttrList function;
        function.set_name(xla_function_info.function_name);
        AddNodeAttr("function", function, &def);
    
        for (Node* node : nodes_to_remove) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/shape_inference.cc

                } else {
                  value.add_int64_val(dim.size());
                }
              }
              AddNodeAttr("value", value, &const_def);
              for (auto const& attr : n->attrs()) {
                if (*attr.first.begin() == '_') {
                  AddNodeAttr(attr.first, attr.second, &const_def);
                }
              }
    
              TF_ASSIGN_OR_RETURN(Node * const_node, graph->AddNode(const_def));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 00:41:19 UTC 2024
    - 13K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/build_xla_ops_pass_test.cc

                                 int num_resource_args, Node** result) {
      NodeDef call_node;
      call_node.set_name(node_name);
      call_node.set_op(callee_name);
      AddNodeAttr(kXlaCompiledKernelAttr, true, &call_node);
      AddNodeAttr(kXlaNumConstantArgsAttr, num_constant_args, &call_node);
      AddNodeAttr(kXlaNumResourceArgsAttr, num_resource_args, &call_node);
      TF_ASSIGN_OR_RETURN(*result, graph->AddNode(call_node));
      return absl::OkStatus();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 12.2K bytes
    - Viewed (0)
Back to top