Searched defs:tensor_sizes (Results 1 – 2 of 2) sorted by relevance
41 auto tensor_sizes = tensor.shape().dim_sizes(); in ConvertTensorToPyArray() local
52 std::vector<int64_t> tensor_sizes(num_spatial_dims + 2, 1); in ExpandWithBatchAndFeatureDimensions() local