- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TF_SetOutput (0.18 sec)
-
tensorflow/c/kernels_test.cc
TF_GetInput(ctx, 3, &input, s); EXPECT_EQ(TF_OUT_OF_RANGE, TF_GetCode(s)); // Copy the input tensor to output. TF_SetOutput(ctx, 0, input, s); EXPECT_EQ(TF_OK, TF_GetCode(s)); TF_SetOutput(ctx, 24, input, s); EXPECT_EQ(TF_OUT_OF_RANGE, TF_GetCode(s)); EXPECT_EQ(TF_UINT8, TF_ExpectedOutputDataType(ctx, 0));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op.cc
TF_TensorBitcastFrom(tensor, k->output_data_type, output, dims, new_dim_count, status); if (TF_GetCode(status) == TF_OK) { TF_SetOutput(ctx, 0, output, status); } delete[] dims; TF_DeleteTensor(output); } if (TF_GetCode(status) != TF_OK) { TF_OpKernelContext_Failure(ctx, status); } TF_DeleteStatus(status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 31 03:28:11 UTC 2021 - 5.9K bytes - Viewed (0) -
tensorflow/c/kernels.h
// Sets the ith output of ctx to tensor. If TF_GetCode(status) is anything but // TF_OK, ctx is left unmodified. // // If i < 0 or i >= TF_NumOutputs(ctx), *status is set to TF_OUT_OF_RANGE. TF_CAPI_EXPORT extern void TF_SetOutput(TF_OpKernelContext* ctx, int i, const TF_Tensor* tensor, TF_Status* status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:46:22 UTC 2024 - 24.6K bytes - Viewed (0) -
tensorflow/c/kernels.cc
CHECK_GE(index, 0); // Crash OK CHECK_LT(index, cc_ctx->num_inputs()); // Crash OK return static_cast<TF_DataType>(cc_ctx->input_dtype(index)); } void TF_SetOutput(TF_OpKernelContext* ctx, int i, const TF_Tensor* tensor, TF_Status* status) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); if (i < 0 || i >= cc_ctx->num_outputs()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0)