Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for instance_key (0.26 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

              {{"y"},
               "CollectiveReduce",
               {"x"},
               {{"T", DT_FLOAT},
                {"group_size", static_cast<int>(devices.size())},
                {"group_key", 0},
                {"instance_key", 0},
                {"merge_op", "Add"},
                {"final_op", "Id"},
                {"subdiv_offsets", std::vector<int>()}},
               /*dep=*/{"assert"}},
          });
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      TFE_OpSetAttrType(op.get(), "T", TFE_TensorHandleDataType(input));
      TFE_OpSetAttrInt(op.get(), "group_size", group_size);
      TFE_OpSetAttrInt(op.get(), "group_key", 0);
      TFE_OpSetAttrInt(op.get(), "instance_key", 0);
      const std::string merge_op("Add");
      TFE_OpSetAttrString(op.get(), "merge_op", merge_op.c_str(),
                          merge_op.length());
      const std::string final_op("Id");
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test_util.cc

      TF_DeleteStatus(status);
    
      TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(in));
      TFE_OpSetAttrInt(op, "group_size", group_size);
      TFE_OpSetAttrInt(op, "group_key", 123);
      TFE_OpSetAttrInt(op, "instance_key", 456);
      TFE_OpSetAttrString(op, "merge_op", "Add", 3);
      TFE_OpSetAttrString(op, "final_op", "Id", 2);
      std::vector<int64_t> subdiv_offsets;
      TFE_OpSetAttrIntList(op, "subdiv_offsets", subdiv_offsets.data(),
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
Back to top