Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TFE_OpSetAttrShape (0.1 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0,
                         status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      auto outputs =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

          if (tensor_shape.unknown_rank()) {
            TFE_OpSetAttrShape(op, attr_name, nullptr, -1, status);
          } else {
            const auto num_dims = tensor_shape.dim_size();
            std::unique_ptr<int64_t[]> dims(new int64_t[num_dims]);
            for (int i = 0; i < num_dims; ++i) {
              dims[i] = tensor_shape.dim(i).size();
            }
            TFE_OpSetAttrShape(op, attr_name, dims.get(), num_dims, status);
          }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top