Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_SetAttrType (0.08 sec)

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

      TF_SetAttrType(placeholder_desc, "dtype", TF_FLOAT);
      TF_Operation* placeholder_op = TF_FinishOperation(placeholder_desc, status);
      if (TF_GetCode(status) != TF_OK) return;
      TF_Output x{placeholder_op, 0};
    
      TF_OperationDescription* reduce_desc =
          TF_NewOperation(body.get(), "CollectiveReduce", "CollectiveReduce");
      TF_SetAttrType(reduce_desc, "T", TF_FLOAT);
    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.cc

      for (int i = 0; i < num_values; ++i) {
        b[i] = values[i];
      }
      desc->node_builder.Attr(attr_name,
                              ArraySlice<const bool>(b.get(), num_values));
    }
    
    void TF_SetAttrType(TF_OperationDescription* desc, const char* attr_name,
                        TF_DataType value) {
      desc->node_builder.Attr(attr_name, static_cast<DataType>(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)
  3. tensorflow/c/c_api_test.cc

        TF_OperationDescription* desc = TF_NewOperation(graph, "Const", name);
        TF_SetAttrTensor(desc, "value", tensor.get(), s);
        if (TF_GetCode(s) != TF_OK) return nullptr;
        TF_SetAttrType(desc, "dtype", TF_FLOAT);
        TF_Operation* op = TF_FinishOperation(desc, s);
        EXPECT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        return op;
      }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
Back to top