Lines Matching defs:shaped
292 /// auto outer = my_ten.shaped<T, 2>({4, 15});
294 /// auto outer = my_ten.shaped<T, 2>({4, 8});
296 /// auto weird = my_ten.shaped<T, 3>({6, 5, 2});
303 return shaped<T, 1>({NumElements()});
335 typename TTypes<T, NDIMS>::Tensor shaped(gtl::ArraySlice<int64> new_sizes);
341 /// The allowed bitcast is the only difference from `shaped()`.
392 return shaped<T, 1>({NumElements()});
401 typename TTypes<T, NDIMS>::ConstTensor shaped(
408 /// The allowed bitcast is the only difference from `shaped()`.
661 typename TTypes<T, NDIMS>::Tensor Tensor::shaped(
688 typename TTypes<T, NDIMS>::ConstTensor Tensor::shaped(
729 return shaped<T, NDIMS>(ComputeFlatInnerDims(shape_.dim_sizes(), NDIMS));
734 return shaped<T, NDIMS>(ComputeFlatOuterDims(shape_.dim_sizes(), NDIMS));
741 return shaped<T, NDIMS>(ComputeFlatInnerDims(flat_outer, NDIMS));
746 return shaped<T, NDIMS>(ComputeFlatInnerDims(shape_.dim_sizes(), NDIMS));
751 return shaped<T, NDIMS>(ComputeFlatOuterDims(shape_.dim_sizes(), NDIMS));
759 return shaped<T, NDIMS>(ComputeFlatInnerDims(flat_outer, NDIMS));
