Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for shape (0.14 sec)

  1. tensorflow/c/c_api.cc

          auto* out_shape_and_type = handle_data.add_shape_and_type();
          ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape());
          out_shape_and_type->set_dtype(p.dtype);
          *out_shape_and_type->mutable_type() = p.type;
        }
      }
      string str_data;
      handle_data.SerializeToString(&str_data);
    
      TF_Buffer* result = TF_NewBufferFromString(str_data.c_str(), str_data.size());
      return result;
    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)
  2. tensorflow/c/c_api_experimental.cc

          shape.dims = nullptr;
          continue;
        }
        shape.dims = new int64_t[shape.num_dims];
        for (size_t j = 0; j < shape.num_dims; ++j) {
          shape.dims[j] = c.Value(c.Dim(shape_handle, j));
        }
      }
      if (output_shapes != nullptr) *output_shapes = output_shapes_result;
    
      // TODO(bgogul): Set output_resource_shapes_and_types.
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_test.cc

      EXPECT_TRUE(
          std::equal(std::begin(partial_shape), std::end(partial_shape), values));
    }
    
    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)};
    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)
Back to top