Searched refs:var_to_shape_map_ (Results 1 – 2 of 2) sorted by relevance
35 var_to_shape_map_(nullptr), in CheckpointReader()49 var_to_shape_map_.swap(result.first); in CheckpointReader()57 var_to_shape_map_.reset( in CheckpointReader()73 CHECK(var_to_shape_map_); in GetVariableToShapeMap()74 return *var_to_shape_map_; in GetVariableToShapeMap()
73 std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map_; variable