Searched refs:tensor_map_ (Results 1 – 3 of 3) sorted by relevance
35 buffer << std::endl << std::string("tensor map = " + tensor_map_.ToString()); in StandardToString()54 tensor_map_ = tensor_map; in Init()109 for (uint64_t i = 0; i < tensor_map_.GetDimSize(); i++) { in TensorShapeDimensionIsDividedBySplitDeviceDimension()110 if (tensor_map_.GetDimByIdx(i) != -1) { in TensorShapeDimensionIsDividedBySplitDeviceDimension()145 (void)tensor_map_.Init(tensor_map_shape); in RemoveElementEqualToOneInDeviceArrangement()151 return tensor_map_.GetIndexByValue(idx); in GetTensorDimensionIndexByDeviceDimensionIndex()156 return static_cast<int64_t>(device_arrangement_.GetDimSize()) - 1 - tensor_map_.GetDimByIdx(idx); in GetSliceDeviceDimensionByTensorDimensionIndex()193 if (tensor_map_.GetIndexByValue(i) < 0) { in ComputeArrangementByExpandedShape()196 re_map_expand_list.push_back((*expand_list_ptr)[LongToUlong(tensor_map_.GetIndexByValue(i))]); in ComputeArrangementByExpandedShape()221 …std::shared_ptr<Map> tensor_map_new_ptr = tensor_map_.ExpandMapByNone(expand_list_pair_ptr->second… in ExpandTensorShapeWithoutExtendDeviceArrangement()[all …]
63 Map tensor_map() const { return tensor_map_; } in tensor_map()81 …bool IsSameTensorMap(const TensorLayout &tensor_layout) const { return (tensor_map_ == tensor_layo… in IsSameTensorMap()141 Map tensor_map_; variable
30 for (auto &impl : tensor_map_) { in ~ModelC()48 std::map<mindspore::tensor::MSTensor *, MSTensor::Impl *> tensor_map_; member in mindspore::ModelC242 auto iter = tensor_map_.find(tensor); in TensorToTensorImpl()243 if (iter != tensor_map_.end()) { in TensorToTensorImpl()251 tensor_map_[tensor] = impl; in TensorToTensorImpl()