- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for TF_ExpectedOutputDataType (0.42 sec)
-
tensorflow/c/kernels/merge_summary_op.cc
return; } *s.add_value() = summary_in.value(v); } } } Safe_TF_TensorPtr summary_tensor(TF_AllocateOutput( /*context=*/ctx, /*index=*/0, /*dtype=*/TF_ExpectedOutputDataType(ctx, 0), /*dims=*/nullptr, /*num_dims=*/0, /*len=*/sizeof(tensorflow::tstring), status.get())); if (TF_GetCode(status.get()) != TF_OK) { TF_OpKernelContext_Failure(ctx, status.get()); return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 31 03:28:11 UTC 2021 - 4.7K bytes - Viewed (0) -
tensorflow/c/kernels/summary_op.cc
v->set_tag(Ttags_i.data(), Ttags_i.size()); v->set_simple_value(static_cast<float>(values_array[i])); } TF_Tensor* summary_tensor = TF_AllocateOutput(ctx, 0, TF_ExpectedOutputDataType(ctx, 0), nullptr, 0, sizeof(tensorflow::tstring), params.status); if (TF_GetCode(params.status) != TF_OK) { TF_DeleteTensor(summary_tensor);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 6.2K bytes - Viewed (0) -
tensorflow/c/kernels/histogram_summary_op.cc
v->set_tag(tag.data(), tag.size()); histo.EncodeToProto(v->mutable_histo(), false /* Drop zero buckets */); Safe_TF_TensorPtr summary_tensor(TF_AllocateOutput( /*context=*/ctx, /*index=*/0, /*dtype=*/TF_ExpectedOutputDataType(ctx, 0), /*dims=*/nullptr, /*num_dims=*/0, /*len=*/sizeof(tensorflow::tstring), status.get())); if (TF_GetCode(status.get()) != TF_OK) { TF_OpKernelContext_Failure(ctx, status.get());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 6.5K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
TF_SetOutput(ctx, 24, input, s); EXPECT_EQ(TF_OUT_OF_RANGE, TF_GetCode(s)); EXPECT_EQ(TF_UINT8, TF_ExpectedOutputDataType(ctx, 0)); EXPECT_DEATH({ TF_ExpectedOutputDataType(ctx, 1); }, "Check failed: i < cc_ctx->num_outputs"); EXPECT_DEATH({ TF_ExpectedOutputDataType(ctx, -1); }, "Check failed: i >= 0"); TF_DeleteStatus(s); if (input != nullptr) {
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.h
// Returns the expected output data type of the ith output. If i < 0 or // i >= TF_NumOutputs(ctx), the program aborts. TF_CAPI_EXPORT extern TF_DataType TF_ExpectedOutputDataType( TF_OpKernelContext* ctx, int i); // Returns true if the ith input is allocated in host memory. If i < 0 or i >= // TF_NumInputs(ctx), the program aborts.
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
TF_StringView string_view_of_name; string_view_of_name.data = cc_ctx->def().name().data(); string_view_of_name.len = cc_ctx->def().name().length(); return string_view_of_name; } TF_DataType TF_ExpectedOutputDataType(TF_OpKernelContext* ctx, int i) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); CHECK_GE(i, 0); CHECK_LT(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)