Searched refs:ByteStridesForShape (Results 1 – 3 of 3) sorted by relevance
174 std::vector<ssize_t> ByteStridesForShape(const Shape& shape) { in ByteStridesForShape() function210 ByteStridesForShape(m.shape()), m.untyped_data(), in LiteralToPython()
131 ByteStridesForShape(buffer_->on_device_shape()), data, in AsNumPyArray()290 extra->strides = ByteStridesForShape(shape); in PjRtBufferGetBuffer()
61 std::vector<ssize_t> ByteStridesForShape(const Shape& shape);