Searched defs:out_strides (Results 1 – 8 of 8) sorted by relevance
37 const int32* out_strides = buf + ndims; in TileKernel() local61 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); in TileSimple() local
32 gtl::InlinedVector<int64, 8> out_strides = ComputeStride<int64>(out->shape()); in TileSimple() local
38 const int32* out_strides = buf + ndims; in TransposeKernel() local66 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); in TransposeSimple() local
39 gtl::InlinedVector<int64, 8> out_strides = ComputeStride<int64>(out->shape()); in TransposeSimple() local
135 memory::dims out_strides = in MKLTransposeND() local
153 size_t out_strides[1]; member
763 size_t out_strides[4]; member
690 size_t out_strides[4]; member