- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for eq (0.15 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_testlib.h
TFE_TensorHandleResolve(handle, status.get()), TF_DeleteTensor); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); ASSERT_EQ(TF_TensorType(actual_value.get()), static_cast<TF_DataType>(DataTypeToEnum<value_type>().value)); EXPECT_EQ(expected_value, *static_cast<value_type*>(TF_TensorData(actual_value.get()))); }
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
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)