Searched refs:ComputeFlatOuterDims (Results 1 – 2 of 2) sorted by relevance
666 static gtl::InlinedVector<int64, 4> ComputeFlatOuterDims(919 return shaped<T, NDIMS>(ComputeFlatOuterDims(shape_.dim_sizes(), NDIMS)); in flat_outer_dims()925 ComputeFlatOuterDims(shape_.dim_sizes(), begin + NDIMS); in flat_inner_outer_dims()936 return shaped<T, NDIMS>(ComputeFlatOuterDims(shape_.dim_sizes(), NDIMS)); in flat_outer_dims()943 ComputeFlatOuterDims(shape_.dim_sizes(), begin + NDIMS); in flat_inner_outer_dims()
1284 gtl::InlinedVector<int64, 4> Tensor::ComputeFlatOuterDims( in ComputeFlatOuterDims() function in tensorflow::Tensor