Searched defs:in_strides (Results 1 – 8 of 8) sorted by relevance
36 const int32* in_strides = buf; in TileKernel() local60 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); in TileSimple() local
31 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); in TileSimple() local
37 const int32* in_strides = buf; in TransposeKernel() local65 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); in TransposeSimple() local
38 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); in TransposeSimple() local
133 memory::dims in_strides = CalculateTFStrides(in_dims); in MKLTransposeND() local
151 size_t in_strides[4]; member
761 size_t in_strides[4]; member
688 size_t in_strides[4]; member