Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 55 for B_dims (0.14 sec)

  1. tensorflow/c/eager/gradient_checker_test.cc

      int64_t A_dims[] = {2, 2};
      AbstractTensorHandlePtr A;
      {
        AbstractTensorHandle* A_raw;
        absl::Status s = TestTensorHandleWithDims<float, TF_FLOAT>(
            ctx_.get(), A_vals, A_dims, 2, &A_raw);
        ASSERT_EQ(errors::OK, s.code()) << s.message();
        A.reset(A_raw);
      }
      float B_vals[] = {.5f, -1.0f, 1.0f, 1.0f};
      int64_t B_dims[] = {2, 2};
      AbstractTensorHandlePtr B;
      {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 6.5K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

      const int num_dims = 2;
      int64_t* dims = new int64_t[num_dims];
      int64_t num_elements = 1;
      dims[0] = batch_size;
      dims[1] = 1;
      for (int64_t i = 0; i < num_dims; ++i) {
        num_elements *= dims[i];
      }
      TF_Tensor* t = TF_AllocateTensor(TF_STRING, dims, num_dims,
                                       sizeof(TF_TString) * num_elements);
      delete[] dims;
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test_util.cc

      constexpr int64_t dims[] = {100, 100};
      constexpr int num_elements = dims[0] * dims[1];
      float data[num_elements];
      for (int i = 0; i < num_elements; ++i) {
        data[i] = 1.0f;
      }
      TF_Status* status = TF_NewStatus();
      TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0],
                                            sizeof(dims) / sizeof(int64_t), status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental_test.cc

      */
    
      // Build an abstract input tensor.
      int64_t dims[] = {2, 2};  // Matrices will be 2 x 2
      int num_dims = sizeof(dims) / sizeof(dims[0]);
    
      float vals[] = {0.0f, 0.0f, 0.0f, 0.0f};
      TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get());
      TFE_TensorHandle* t =
          TestMatrixTensorHandleWithInput(eager_ctx, vals, dims, num_dims);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 39.1K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental.cc

        std::vector<DimensionHandle> dims;
        const TF_ShapeAndType& input_shape = input_shapes->items[i];
        if (input_shape.num_dims == InferenceContext::kUnknownRank) {
          c.SetInput(i, c.UnknownShape());
          continue;
        }
        dims.reserve(input_shape.num_dims);
        for (int j = 0; j < input_shape.num_dims; ++j) {
          dims.push_back(c.MakeDim(input_shape.dims[j]));
        }
        c.SetInput(i, c.MakeShape(dims));
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_experimental_test.cc

        int num_dims = output_shapes->items[0].num_dims;
        int64_t* dims = output_shapes->items[0].dims;
    
        if (!expected_shape.has_value()) {
          EXPECT_EQ(num_dims, -1);
          EXPECT_EQ(dims, nullptr);
          return;
        }
    
        EXPECT_EQ(num_dims, expected_shape->size());
        for (size_t i = 0; i < num_dims; ++i) {
          EXPECT_EQ(dims[i], (*expected_shape)[i]);
        }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Jan 17 22:27:52 UTC 2023
    - 13.1K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_unified_experimental_graph.cc

          return absl::OkStatus();
        }
    
        std::vector<int64_t> dims(num_dims, kUnknownDim);
        TF_GraphGetTensorShape(graph_, output_,
                               reinterpret_cast<int64_t*>(dims.data()), num_dims,
                               &status);
        TF_RETURN_IF_ERROR(StatusFromTF_Status(&status));
        TF_RETURN_IF_ERROR(tensorflow::TensorShapeUtils::MakeShape(dims, shape));
    
        return absl::OkStatus();
      }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  8. tensorflow/c/c_test_util.cc

    }
    
    TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values) {
      int64_t num_values = 1;
      for (int i = 0; i < num_dims; ++i) {
        num_values *= dims[i];
      }
      TF_Tensor* t =
          TF_AllocateTensor(TF_INT8, dims, num_dims, sizeof(char) * num_values);
      memcpy(TF_TensorData(t), values, sizeof(char) * num_values);
      return t;
    }
    
    TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api_test_util.h

                                                      float data[], int64_t dims[],
                                                      int num_dims);
    
    // Get a Matrix TensorHandle with given float values and dimensions
    TFE_TensorHandle* TestTensorHandleWithDimsFloat(TFE_Context* ctx, float data[],
                                                    int64_t dims[], int num_dims);
    
    // Get a Matrix TensorHandle with given int values and dimensions
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Jul 17 23:43:59 UTC 2023
    - 7.7K bytes
    - Viewed (0)
  10. tensorflow/c/c_test_util.h

    TF_Tensor* BoolTensor(int32_t v);
    
    // Create a tensor with values of type TF_INT8 provided by `values`.
    TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values);
    
    // Create a tensor with values of type TF_INT32 provided by `values`.
    TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims,
                           const int32_t* values);
    
    // Create 1 dimensional tensor with values from `values`
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 09 01:06:53 UTC 2018
    - 6K bytes
    - Viewed (0)
Back to top