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