Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for node_def (0.1 sec)

  1. tensorflow/c/c_api_experimental.cc

      node_def.set_name(op->Name());
      node_def.set_op(op->Name());
      for (int i = 0; i < num_inputs; ++i) {
        node_def.add_input("dummy_input");
      }
      OperationFromInterface(op)->Attrs().FillAttrValueMap(node_def.mutable_attr());
    
      const tensorflow::OpRegistrationData* op_reg_data;
      status->status =
          tensorflow::OpRegistry::Global()->LookUp(node_def.op(), &op_reg_data);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  2. tensorflow/c/c_test_util.cc

          }
        }
      }
      return found_dtype && found_value;
    }
    
    bool IsAddN(const tensorflow::NodeDef& node_def, int n) {
      if (node_def.op() != "AddN" || node_def.name() != "add" ||
          node_def.input_size() != n) {
        return false;
      }
      bool found_t = false;
      bool found_n = false;
      for (const auto& attr : node_def.attr()) {
        if (attr.first == "T") {
          if (attr.second.type() == tensorflow::DT_INT32) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_distributed_test.cc

          "      }"
          "    }"
          "    node_def {"
          "      name: 'read0'"
          "      op: 'ReadVariableOp'"
          "      input: 'var'"
          "      device: '/job:localhost/replica:0/task:0/device:CPU:0'"
          "      attr {"
          "        key: 'dtype'"
          "        value {"
          "          type: DT_FLOAT"
          "        }"
          "      }"
          "    }"
          "    node_def {"
          "      name: 'read1'"
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 23.4K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_function.cc

    #include "tensorflow/core/framework/function.h"
    #include "tensorflow/core/framework/function.pb.h"
    #include "tensorflow/core/framework/graph_to_functiondef.h"
    #include "tensorflow/core/framework/node_def.pb.h"
    #include "tensorflow/core/framework/node_def_util.h"
    #include "tensorflow/core/framework/tensor.pb.h"  // NOLINT
    #include "tensorflow/core/framework/types.h"
    #include "tensorflow/core/graph/graph.h"
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 13.7K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental.h

                                                         const TF_DataType* values,
                                                         int num_values);
    
    // Checks the tensorflow::NodeDef built via the methods above to see if it can
    // run on device_type.
    TF_CAPI_EXPORT extern void TF_AttrBuilderCheckCanRunOnDevice(
        TF_AttrBuilder* builder, const char* device_type, TF_Status* status);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
Back to top