- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for input_shape (0.05 sec)
-
tensorflow/c/c_api_experimental_test.cc
- // Create input_shapes.
- TF_ShapeAndTypeList* input_shapes =
- TF_NewShapeAndTypeList(input_shapes_vec.size());
- for (size_t i = 0; i < input_shapes_vec.size(); ++i) {
- const auto& input_shape = input_shapes_vec[i];
- if (input_shape.has_value()) {
- TF_ShapeAndTypeListSetShape(input_shapes, i, input_shape->data(),
- input_shape->size());
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Tue Jan 17 22:27:52 UTC 2023 - 13.1K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
- // Set input_shapes.
- for (int i = 0; i < num_inputs; ++i) {
- 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) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 29.5K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
- // if you want shape inference to consider the input tensors of the
- // op for shape inference.
- // - The types need not be set in `input_shapes` as it is not used.
- // - The number of `input_tensors` should be the same as the number of items
- // in `input_shapes`.
- //
- // The results are returned in `output_shapes` and
- // `output_resource_shapes_and_types`. The caller is responsible for freeing the
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 15.1K bytes - Viewed (0)