- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for eq (0.17 sec)
-
tensorflow/c/eager/c_api_test_util.h
TF_Tensor* t = TFE_AllocateHostTensor(ctx, datatype, dims, num_dims, status); memcpy(TF_TensorData(t), data, TF_TensorByteSize(t)); TFE_TensorHandle* th = TFE_NewTensorHandleFromTensor(ctx, t, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TF_DeleteTensor(t); TF_DeleteStatus(status); return th; } // Return a scalar tensor handle with given values. template <class T, TF_DataType datatype>
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Jul 17 23:43:59 GMT 2023 - 7.7K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
bool GradientTape<Gradient, BackwardFunction, TapeTensor>::ShouldRecord( absl::Span<const int64_t> tensor_ids, absl::Span<const tensorflow::DataType> dtypes) const { CHECK_EQ(tensor_ids.size(), dtypes.size()); for (int i = 0; i < tensor_ids.size(); ++i) { if (tensor_tape_.find(tensor_ids[i]) != tensor_tape_.end()) { if (IsDtypeTrainable(dtypes[i])) { return true; }
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 02 12:40:29 GMT 2024 - 47.2K bytes - Viewed (1)