Searched refs:ShapeArray (Results 1 – 2 of 2) sorted by relevance
42 using ShapeArray = sparse::SparseTensor::ShapeArray; typedef53 Status GroupShape(const VarDimArray& input_shape, ShapeArray* grouped_shape) { in GroupShape()60 *grouped_shape = ShapeArray(input_shape.begin(), input_shape.end() - 1); in GroupShape()119 const ShapeArray Strides(const VarDimArray& shape) { in Strides()120 ShapeArray result(shape.size()); in Strides()258 ShapeArray output_shape; in Compute()392 ShapeArray* group_shape) { in GroupShapeFromInputs()393 ShapeArray group_shape_1; in GroupShapeFromInputs()395 ShapeArray group_shape_2; in GroupShapeFromInputs()416 ShapeArray TensorShapeToArray(const TensorShape& t) { in TensorShapeToArray()[all …]
45 typedef typename gtl::InlinedVector<int64, 8> ShapeArray; typedef275 static inline ShapeArray UndefinedOrder(const VarDimArray shape) { in UndefinedOrder()276 return ShapeArray(shape.size(), -1); in UndefinedOrder()279 static inline ShapeArray TensorShapeToVector(const TensorShape& shape) { in TensorShapeToVector()280 ShapeArray vec(shape.dims()); in TensorShapeToVector()371 ShapeArray shape_;372 ShapeArray order_;435 order_ = ShapeArray(order.begin(), order.end()); in Reorder()507 ShapeArray final_order(order_0.begin(), order_0.end()); in Concat()508 ShapeArray final_shape(tensors[0].shape().begin(), tensors[0].shape().end()); in Concat()