Searched refs:ComputeStride (Results 1 – 5 of 5) sorted by relevance
30 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); in TileSimpleImpl()31 gtl::InlinedVector<int64, 8> out_strides = ComputeStride<int64>(out->shape()); in TileSimpleImpl()
60 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); in TileSimple()61 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); in TileSimple()
38 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); in TransposeSimple()39 gtl::InlinedVector<int64, 8> out_strides = ComputeStride<int64>(out->shape()); in TransposeSimple()
66 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); in TransposeSimple()67 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); in TransposeSimple()
89 gtl::InlinedVector<T, 8> ComputeStride(const TensorShape& shape) { in ComputeStride() function