Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TF_NewDimensionHandle (0.26 sec)

  1. tensorflow/c/kernels/ops/bitcast.cc

      if (input_type_size < output_type_size) {
        TF_ShapeInferenceContextWithRankAtLeast(ctx, shape, 1, shape, status);
    
        if (TF_GetCode(status) == TF_OK) {
          TF_DimensionHandle* last_dim = TF_NewDimensionHandle();
          size_t divisor_val = output_type_size / input_type_size;
          TF_ShapeInferenceContextDim(ctx, shape, -1, last_dim);
          if (!TF_DimensionHandleValueKnown(last_dim) ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 07:51:50 UTC 2024
    - 5.1K bytes
    - Viewed (0)
  2. tensorflow/c/ops.cc

                                     *reinterpret_cast<ShapeHandle*>(second),
                                     reinterpret_cast<ShapeHandle*>(result));
      Set_TF_Status_from_Status(status, s);
    }
    
    TF_DimensionHandle* TF_NewDimensionHandle() {
      return reinterpret_cast<TF_DimensionHandle*>(new DimensionHandle);
    }
    
    int64_t TF_ShapeInferenceContextNumInputs(TF_ShapeInferenceContext* ctx) {
      auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 28 22:41:35 UTC 2022
    - 10.9K bytes
    - Viewed (0)
  3. tensorflow/c/ops_test.cc

      ASSERT_EQ(1, TF_ShapeInferenceContextRankKnown(C_CTX(&c), handle));
      ASSERT_EQ(1, TF_ShapeInferenceContextRank(C_CTX(&c), handle));
    
      TF_DimensionHandle* dim_handle = TF_NewDimensionHandle();
      TF_ShapeInferenceContextDim(C_CTX(&c), handle, 0, dim_handle);
      ASSERT_EQ(1, TF_DimensionHandleValueKnown(dim_handle));
      ASSERT_EQ(43, TF_DimensionHandleValue(dim_handle));
      TF_DeleteShapeHandle(handle);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 11 01:20:50 UTC 2021
    - 12.6K bytes
    - Viewed (0)
  4. tensorflow/c/ops.h

        TF_ShapeInferenceContext* ctx, size_t size);
    
    // Returns a newly allocated dimension handle. It must be freed with
    // TF_DeleteDimensionHandle.
    TF_CAPI_EXPORT extern TF_DimensionHandle* TF_NewDimensionHandle();
    
    // Interprets the named shape inference context attribute as a TF_DataType and
    // places it into *val. *status is set to TF_OK.
    //
    // If the attribute could not be found or could not be interpreted as
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 16.3K bytes
    - Viewed (0)
Back to top