Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_Dim (0.19 sec)

  1. tensorflow/c/experimental/gradients/grad_test_helper.cc

      int64_t num_elem_analytical = 1;
      auto num_dims_analytical = TF_NumDims(analytical_tensor);
      ASSERT_EQ(dims.size(), num_dims_analytical);
      for (int j = 0; j < num_dims_analytical; j++) {
        auto dim_analytical = TF_Dim(analytical_tensor, j);
        ASSERT_EQ(dims[j], dim_analytical);
        num_elem_analytical *= dim_analytical;
      }
    
      float* danalytical = new float[num_elem_analytical]{0};
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradient_checker.cc

      }
    }
    
    // Fills out_dims with the dimensions of the given tensor.
    void GetDims(const TF_Tensor* t, int64_t* out_dims) {
      int num_dims = TF_NumDims(t);
      for (int i = 0; i < num_dims; i++) {
        out_dims[i] = TF_Dim(t, i);
      }
    }
    
    // Runs model as is if output is a scalar,
    // else sums the output tensor before returning.
    Status RunAndMaybeSum(AbstractContext* ctx, Model forward,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7.3K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_test.cc

                                  &Deallocator, &deallocator_called);
      EXPECT_FALSE(deallocator_called);
      EXPECT_EQ(TF_FLOAT, TF_TensorType(t));
      EXPECT_EQ(2, TF_NumDims(t));
      EXPECT_EQ(dims[0], TF_Dim(t, 0));
      EXPECT_EQ(dims[1], TF_Dim(t, 1));
      EXPECT_EQ(num_bytes, TF_TensorByteSize(t));
      EXPECT_EQ(static_cast<void*>(values), TF_TensorData(t));
      TF_DeleteTensor(t);
      EXPECT_TRUE(deallocator_called);
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
Back to top