Home
last modified time | relevance | path

Searched refs:LayoutShapeForChannel (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.h253 Shape LayoutShapeForChannel(Shape shape, int64 channel_id) const { in LayoutShapeForChannel() function
Dlayout_assignment.cc505 ->LayoutShapeForChannel(send_buffer_shape, in AddMandatoryConstraints()
519 ->LayoutShapeForChannel(recv_buffer_shape, in AddMandatoryConstraints()
540 ->LayoutShapeForChannel(buffer_shape, channel_id); in AddMandatoryConstraints()