Searched refs:send_buffer_shape (Results 1 – 1 of 1) sorted by relevance
501 const Shape send_buffer_shape = instruction->operand(0)->shape(); in AddMandatoryConstraints() local502 TF_RET_CHECK(send_buffer_shape.IsArray()); in AddMandatoryConstraints()505 ->LayoutShapeForChannel(send_buffer_shape, in AddMandatoryConstraints()