Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for TF_ExpectedOutputDataType (0.42 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
Back to top