- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for tensor_dtypes (0.31 sec)
-
tensorflow/c/eager/gradients.cc
std::vector<int64_t> tensor_ids(tensors.size()); std::vector<tensorflow::DataType> tensor_dtypes(tensors.size()); for (int i = 0; i < tensors.size(); i++) { tensor_ids[i] = ToId(tensors[i]); tensor_dtypes[i] = tensors[i]->DataType(); } return GradientTape::ShouldRecord(tensor_ids, tensor_dtypes); } void Tape::DeleteTrace(const AbstractTensorHandle* t) { GradientTape::DeleteTrace(ToId(t)); }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, op_name.c_str(), status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TF_DeleteStatus(status); TFE_OpSetAttrType(op, "tensor_type", TF_INT32); TFE_OpSetAttrString(op, "tensor_name", "dummy", 5); TFE_OpSetAttrString(op, "send_device", send_device.c_str(), send_device.size());
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0)