Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for shape_1 (0.07 sec)

  1. tensorflow/c/c_api_test.cc

    TEST_F(CApiAttributesTest, ShapeList) {
      const int64_t shape_1[] = {1, 3};
      const int64_t shape_2[] = {2, 4, 6};
      const int64_t* list[] = {&shape_1[0], &shape_2[0]};
      const size_t list_size = TF_ARRAYSIZE(list);
      const int ndims[] = {TF_ARRAYSIZE(shape_1), TF_ARRAYSIZE(shape_2)};
      const int total_ndims = 5;  // ndims[0] + ndims[1]
    
      auto desc = init("list(shape)");
      TF_SetAttrShapeList(desc, "v", list, ndims, list_size);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.h

    // setting a shape of [-1, 2] with an existing shape [2, -1] would set
    // a final shape of [2, 2] based on shape merging semantics.
    //
    // Returns an error into `status` if:
    //   * `output` is not in `graph`.
    //   * An invalid shape is being set (e.g., the shape being set
    //     is incompatible with the existing shape).
    TF_CAPI_EXPORT extern void TF_GraphSetTensorShape(TF_Graph* graph,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test.cc

        ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
    
        TFE_Op* shape_op = ShapeOp(ctx, hgpu);
        TFE_OpSetDevice(shape_op, gpu_device_name.c_str(), status.get());
        ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
        TFE_TensorHandle* retvals[1];
        int num_retvals = 1;
        TFE_Execute(shape_op, &retvals[0], &num_retvals, status.get());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_function_test.cc

                                                               TF_DeleteStatus);
    
      TF_Tensor* tensor_shape = Int32Tensor({37, 1});
      TF_Operation* shape = Const(tensor_shape, func_graph.get(), s.get(), "shape");
      TF_Operation* random =
          RandomUniform(shape, TF_FLOAT, func_graph.get(), s.get());
    
      TF_Output outputs[] = {{random, 0}};
      *func = TF_GraphToFunction(func_graph.get(), name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Jul 20 22:08:54 UTC 2023
    - 63.6K bytes
    - Viewed (0)
Back to top