- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TF_SetAttrType (0.08 sec)
-
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) -
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) -
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)