| 104 | } |
| 105 | |
| 106 | TFE_TensorHandle* DoubleTestMatrixTensorHandle3X2() { |
| 107 | int64_t dims[] = {3, 2}; |
| 108 | double data[] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; |
| 109 | TF_Tensor* t = TF_AllocateTensor( |
| 110 | TF_FLOAT, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data)); |
| 111 | memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); |
| 112 | TF_Status* status = TF_NewStatus(); |
| 113 | TFE_TensorHandle* th = TFE_NewTensorHandle(t, status); |
| 114 | CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); |
| 115 | TF_DeleteTensor(t); |
| 116 | TF_DeleteStatus(status); |
| 117 | return th; |
| 118 | } |
| 119 | |
| 120 | TFE_TensorHandle* TestMatrixTensorHandle3X2() { |
| 121 | int64_t dims[] = {3, 2}; |
no test coverage detected