Searched refs:ComputeStride (Results 1 – 5 of 5) sorted by relevance
31 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); in TileSimple()32 gtl::InlinedVector<int64, 8> out_strides = ComputeStride<int64>(out->shape()); in TileSimple()
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()
65 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); in TransposeSimple()66 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); in TransposeSimple()
90 gtl::InlinedVector<T, 8> ComputeStride(const TensorShape& shape) { in ComputeStride() function