Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 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)
  4. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

        AddNodeAttr("shape_inference_graph", shape_inference_graph, node_def);
        AddNodeAttr("shapes", std::vector<TensorShapeProto>{}, node_def);
      }
      AddNodeAttr("ancestors", std::vector<string>{}, node_def);
      AddNodeAttr("Tinputs", recv_at_host_dtypes, node_def);
      AddNodeAttr("Toutputs", send_from_host_dtypes, node_def);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

            // outputs into host-memory constants and device-memory non-constants.
    
            AddNodeAttr(kXlaCompiledKernelAttr, true, node);
            AddNodeAttr(kXlaNumConstantArgsAttr, num_consts, node);
            AddNodeAttr(kXlaNumResourceArgsAttr, num_resources, node);
            return absl::OkStatus();
          };
    
      TF_RETURN_WITH_CONTEXT_IF_ERROR(
    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/mlir/tensorflow/translate/import_model.cc

      }
      // Update op name, drop inputs and set attributes required by the Placeholder
      // op.
      *node->mutable_op() = "Placeholder";
      node->clear_attr();
      node->clear_input();
      AddNodeAttr("dtype", dtype, node);
      AddNodeAttr("shape", it->second.shape, node);
      return absl::OkStatus();
    }
    
    // Preprocesses GraphDef before it can be converted to Graph by,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top