Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_SetAttrTensorShapeProto (0.61 sec)

  1. tensorflow/c/c_api.h

                                                   int num_shapes);
    // `proto` must point to an array of `proto_len` bytes representing a
    // binary-serialized TensorShapeProto.
    TF_CAPI_EXPORT extern void TF_SetAttrTensorShapeProto(
        TF_OperationDescription* desc, const char* attr_name, const void* proto,
        size_t proto_len, TF_Status* status);
    // `protos` and `proto_lens` must point to arrays of length `num_shapes`.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
  2. tensorflow/c/c_api_test.cc

      tensorflow::TensorShapeProto proto;
      tensorflow::PartialTensorShape(pts).AsProto(&proto);
      string bytes;
      proto.SerializeToString(&bytes);
    
      auto desc = init("shape");
      TF_SetAttrTensorShapeProto(desc, "v", bytes.data(), bytes.length(), s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
    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)
  3. tensorflow/c/c_api.cc

        } else {
          shapes.emplace_back(ArraySlice<int64_t>(
              reinterpret_cast<const int64_t*>(dims[i]), num_dims[i]));
        }
      }
      desc->node_builder.Attr(attr_name, shapes);
    }
    
    void TF_SetAttrTensorShapeProto(TF_OperationDescription* desc,
                                    const char* attr_name, const void* proto,
                                    size_t proto_len, TF_Status* status) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
Back to top