Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_SetAttrInt (0.07 sec)

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

      TF_OperationDescription* reduce_desc =
          TF_NewOperation(body.get(), "CollectiveReduce", "CollectiveReduce");
      TF_SetAttrType(reduce_desc, "T", TF_FLOAT);
      TF_SetAttrInt(reduce_desc, "group_size", group_size);
      TF_SetAttrInt(reduce_desc, "group_key", 0);
      TF_SetAttrInt(reduce_desc, "instance_key", 0);
    
      const std::string merge_op("Mul");
      TF_SetAttrString(reduce_desc, "merge_op", merge_op.c_str(),
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

        EXPECT_EQ(list[i], string(static_cast<const char*>(values[i]), lens[i]))
            << i;
      }
    }
    
    TEST_F(CApiAttributesTest, Int) {
      auto desc = init("int");
      TF_SetAttrInt(desc, "v", 31415);
    
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_TF_META("v", -1, TF_ATTR_INT, -1);
    
      int64_t value;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  3. tensorflow/c/c_api.cc

        v.reserve(num_values);
        for (int i = 0; i < num_values; ++i) {
          v.emplace_back(static_cast<const char*>(values[i]), lengths[i]);
        }
        desc->node_builder.Attr(attr_name, v);
      }
    }
    
    void TF_SetAttrInt(TF_OperationDescription* desc, const char* attr_name,
                       int64_t value) {
      desc->node_builder.Attr(attr_name, static_cast<int64_t>(value));
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top