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