Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for var_to_shape_map (0.16 sec)

  1. tensorflow/c/checkpoint_reader.cc

            << entry.InitializationErrorString();
        string key(v2_reader_->key());
        (*var_to_shape_map)[key] = TensorShape(entry.shape());
        (*var_to_data_type_map)[key] = DataType(entry.dtype());
      }
      // The returned pointers are owned by the caller.
      return std::make_pair(std::move(var_to_shape_map),
                            std::move(var_to_data_type_map));
    }
    
    }  // namespace checkpoint
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 5.6K bytes
    - Viewed (0)
Back to top