Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TF_SetAttrInt (0.04 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/c/c_test_util.cc 

          ScalarConst(0, graph, s, absl::StrCat(name, "_const0").c_str());
  TF_OperationDescription* desc = TF_NewOperation(graph, "Split", name);
  TF_AddInput(desc, {zero, 0});
  TF_AddInput(desc, {input, 0});
  TF_SetAttrInt(desc, "num_split", 3);
  TF_SetAttrType(desc, "T", TF_INT32);
  // Set device to CPU since there is no version of split for int32 on GPU
  // TODO(iga): Convert all these helpers and tests to use floats because
    Registered: Tue Dec 30 12:39:10 UTC 2025
    - Last Modified: Sat Oct 04 05:55:32 UTC 2025
    - 17.8K bytes
    - Viewed (1)
Back to top