Searched defs:tmp_shape (Results 1 – 9 of 9) sorted by relevance
153 std::vector<int64_t> tmp_shape; in UpdateInputShapeAndType() local188 std::vector<int64_t> tmp_shape; in UpdateOutputShapeAndType() local
60 vvector tmp_shape; in TensorListReserveInferShape() local
102 std::vector<int> tmp_shape = input_tensors_.at(0)->shape(); in CalculateInnerOuterSize() local
138 auto tmp_shape = in_tensors_.at(0)->shape(); in CalculateInnerOuterSize() local
137 … auto tmp_shape = input_shape[idx_i + PADDING_SHAPE_1] + paddings_[idx_i][0] + paddings_[idx_i][1]; in CheckParam() local
147 …auto tmp_shape = input_shape[idx_i + CROPS_SHAPE_1] * block_size_ - crops_[idx_i][0] - crops_[idx_… in CheckParam() local
279 std::vector<dsize_t> tmp_shape = {}; in ColumnLoad() local
464 auto tmp_shape = x->shape()->shape(); in InferImplAllGather() local488 auto tmp_shape = x->shape()->shape(); in InferImplReduceScatter() local
139 ShapeVector tmp_shape = {dout_shape[0], dout_shape[2], dout_shape[3], dout_shape[1]}; in Conv2DBackpropInputInferShape() local